1. 程式人生 > WINDOWS開發 >win10/windows 安裝Pytorch

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