之前基本上都用Keras评也，最近越來越多的聽說PyTorch逐样，就想著也學習一下，順便把學習的過程記錄下來汤徽。在便于以后自己查閱的同時娩缰，希望也能幫到有需要的人吧。
本文介紹了我的安裝方法以及安裝完成后的測試谒府。
1.安裝PyTorch
首先拼坎，去PyTorch官網(wǎng)查看PyTorch的安裝方法。打開頁面后狱掂，我們看到PyTorch官方給我們提供了自動生成安裝命令的選項卡演痒，如下圖所示。
image
根據(jù)自己的實際情況選擇就行趋惨，沒有支持CUDA的GPU的話鸟顺，在
CUDA那一項選None即可。系統(tǒng)安裝了CUDA器虾，但不知道CUDA版本的話讯嫂，運行以下命令查看：
cat /usr/local/cuda/version.txt
選定后，得到了我的安裝命令：
pip3 install torch torchvision
如果網(wǎng)絡慢的話兆沙，可以采用清華的鏡像欧芽，在命令后加上-i https://pypi.tuna.tsinghua.edu.cn/simple some-package即可，即：
pip3 install torch torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple
2.測試安裝是否成功
PyTorch官網(wǎng)給了測試安裝是否成功的例子葛圃，在python中執(zhí)行來進行測試即可：
from __future__ import print_function
import torch
x = torch.rand(5, 3)
print(x)
如果輸出類似下圖的五行三列矩陣千扔，就說明安裝成功了。
image
版權(quán)聲明：本文為博主原創(chuàng)文章库正，遵循 CC 4.0 BY 版權(quán)協(xié)議曲楚，轉(zhuǎn)載請附上原文出處鏈接和本聲明。
本文鏈接：https://ywsun.site/articles/24.html
