寫在前面:
我之前寫過一篇Pytorch的安裝教程:https://blog.csdn.net/qq_33302004/article/details/106320649
但由於之前不太了Pytorch和GPU的關係,所以安裝了錯誤的cuda版本,導致後期無法使用GPU進行計算,所以進行了重裝。在重裝過程中我發現,在終端中使用指令進行安裝速度太慢(試了各種版本換源、換手機熱點),所以探索了離線安裝的方法。
本文內容包括:查看電腦的CUDA版本;離線安裝Pytorch;測試GPU是否可用
查看電腦的CUDA版本:
1.在桌面點擊鼠標右鍵進入NVIDIA控制面板
2.幫助->系統信息
3.點擊組件,可以看到我的電腦支持CUDA 10.1,而我當時選擇了10.2(淚奔)
到這裏你就知道了自己的CUDA版本,然後按照這篇教程安裝即可,CUDA裏面選則你對應的版本。
離線安裝Pytorch:
1.進入官網:https://pytorch.org/。選擇你的版本,然後複製後面那段網址,我這裏的是:
https://download.pytorch.org/whl/torch_stable.html
2.用瀏覽器打開,這裏就會出現很多文件,我這裏是要下載CUDA10.1版本的,你就往下拉,找到cu101開頭的文件,目前最新版是1.5.0,就找對應的版本。
3.上圖中cp後面的數字代表你的python版本,所以你可以查看一下你的python是幾點幾,打開終端輸入python即可,可以看到我的是3.6。
4.所以我選擇cp36的文件,windows系統,所以選擇win_amd64,也就是這個:
5.用迅雷下載。下載好之後利用終端進入下載文件夾,輸入pip install 文件名即可安裝。安裝好後再將官網中生成的那個指令,再終端輸入,因爲安裝過torch了,所以只會安裝torchvision,很快的。
測試GPU:
新建一個python文件,輸入如下代碼:
import torch
print(torch.__version__)
print(torch.version.cuda)
a = torch.cuda.is_available()
print(a)
ngpu= 1
# Decide which device we want to run on
device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
print(device)
print(torch.cuda.get_device_name(0))
print(torch.rand(3,3).cuda())
輸出如下,True表示GPU可用