https://pytorch.org/get-started/locally/
去官網,選擇你需要的版本。
把
pip install torch==1.5.0+cu101 torchvision==0.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
命令行執行。
CUDA(Compute Unified Device Architecture),是顯卡廠商NVIDIA推出的運算平台。查看自己的CUDA版本步驟如下:
1.打開控制面板
在按住Win鍵的情況下,按Q鍵,呼喚出搜索框。
在搜索框中輸入control panel
,如下圖所示:

image.png
2.進入NVIDIA控制面板
在上圖的搜索框中,已經顯示出NVIDIA控制面板,如果讀者有顯示,則可以忽略第一步,直接點擊進入NVIDIA控制面板。
下面2張圖演示如何在控制面板中找出NVIDIA控制面板。
首先選擇查看方式為小圖標,如下圖紅色方框標注處所示:

image.png
然后在小圖標中找出 NVIDIA控制面板,如下圖紅色方框標注處所示:

image.png
3.查看系統信息
選擇進入導航欄的幫助中的系統信息,如下圖紅色方框標注處所示:

image.png
4.查看組件
點擊組件按鈕,如下圖紅色箭頭標注處所示。
在紅色方框和紅色文字共同標注處則顯示當前電腦的CUDA版本號,例如本文作者是CUDA9.2。

image.png
5.總結
1.2018年9月25日是中秋節假后三天第1天上班,寫點簡單的新手指南恢復學習狀態。
2.知道自己電腦的CUDA版本號,則可以 選擇合適版本的CUDA Toolkit,例如上圖本文作者的CUDA版本號為9.2,則我可以安裝CUDA Toolkit 9.2、CUDA Toolkit 9.0、CUDA Toolkit 9.1、CUDA Toolkit 8.0等。
3.查看CUDA Toolkit歷史版本官方鏈接:https://developer.nvidia.com/cuda-toolkit-archive
作者:瀟灑坤
鏈接:https://www.jianshu.com/p/d3b9419a0f89
來源:簡書
著作權歸作者所有。商業轉載請聯系作者獲得授權,非商業轉載請注明出處。