Pytorch離線安裝 —— Pytorch的卸載重裝之旅

寫在前面:

我之前寫過一篇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可用

 

 

 

 

 

 

 

 

 

 

 

 

 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章