1. 程式人生 > 程式設計 >win10從零安裝配置pytorch全過程圖文詳解

win10從零安裝配置pytorch全過程圖文詳解

1.安裝anaconda

(anaconda內建python在內的許多package,所以不用另外下載python)
可以點選下面的清華開源軟體映象站,在官網下載anaconda不如在這下的快
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/映象站

我下載的是下圖這個版本,對應的是3.7,anaconda3代表對應python3

我下載的是這個版本在這裡插入圖片描述

下載過程中基本上都是預設選項

在這裡插入圖片描述

在這裡插入圖片描述

在這裡插入圖片描述

在這裡插入圖片描述

下面這步可以不勾選add anaconda to my PATH environment variable,事後可以再新增

在這裡插入圖片描述

如果有下面這步,記得跳過,我這個版本的anaconda是沒有這步的

在這裡插入圖片描述

在這裡插入圖片描述

下面檢驗一下安裝成功了沒有,在開始開啟anaconda prompt,看到如圖所示即成功

在這裡插入圖片描述

在這裡插入圖片描述

2.確認顯示卡及及時更新(為之後的pytorch安裝作準備)

開啟工作管理員-效能,檢視有GPU還是隻有集顯

在這裡插入圖片描述

再win+r,執行cmd,輸入nvidia-smi後回車

我這裡發生如下報錯,在環境變數path裡面新增C:\Program Files\NVIDIA Corporation\NVSMI 即可解決

在這裡插入圖片描述

正常執行得到的

在這裡插入圖片描述

這裡需要注意的是,表中的Driver Version需大於396.26,否則就去geforce experience更新顯示卡吧,我這裡更新了最新版。

3.下載pytorch


https://pytorch.org/官網
這裡需要注意的是,如果之前第二步檢視時有GPU,則CUDA選擇10即可,如果選9,之後不好用映象源下載,速度會很慢;如果只有集顯,那麼CUDA選擇None即可

在這裡插入圖片描述

之後我們複製Run this Command裡的安裝指令

在這裡插入圖片描述

開啟anaconda prompt,輸入下列指令:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/main/

conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/free/

既然選擇了用映象源下載,那麼安裝指令的 -c pytorch需刪去,也就是:conda install pytorch torchvision cudatoolkit=10.1

這裡我們建立虛擬環境conda create -n your_env_name(這裡就是你給虛擬環境取的名字,筆者取名pytorch) python=X.X(筆者選下3.6),之後在裡面操作,輸入指令conda install pytorch torchvision cudatoolkit=10.1,然後y(es)

在這裡插入圖片描述
在這裡插入圖片描述

最後我們檢驗安裝是否成功,先鍵入python進入python工作環境,然後鍵入import torch,如果未報錯,則pytorch安裝成功,之後輸入指令torce.cuda.is_available(),如果顯示Turn,則GPU可被pytorch使用

在這裡插入圖片描述

4.pycharm配置

開啟pycharm,檔案-新專案-Pure Python
Location自行選擇,Interpreter選擇開啟你安裝anaconda的位置,選擇envs/your_env_name(虛擬環境名)/python.exe即可

在這裡插入圖片描述

建立後 在python console處執行再次執行import torch,無報錯後執行torch.cuda.is_available(),如果顯示Turn,則配置

總結

到此這篇關於win10從零安裝配置pytorch全過程的文章就介紹到這了,更多相關window10 pytorch安裝內容請搜尋我們以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援我們!