win10/windows 安裝Pytorch


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
來源:簡書
著作權歸作者所有。商業轉載請聯系作者獲得授權,非商業轉載請注明出處。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM