windows下配置pytorch環境


借鑒了B站大佬的視頻,自己總結安裝如下。

首先安裝anaconda

按照操作順序,依次安裝,按照我個人習慣,不喜歡講文件安裝在C盤,你們自己決定。

安裝完畢之后。

之后打開Anaconda Prompt,如下圖:

換源:

輸入紅線命令到Anaconda Prompt中,不過我建議換清華源,因為用conda安裝軟件時,比如安裝pytroch就會很容易中斷,因為太大了。換源命令如下:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

  

 之后創建新的pytorch賬戶,名字隨便起。我寫的是pytorch

 

表示創建一個pytorch賬戶,-n表示名字,python=3.6表示這個環境所用的包。

按回車輸入y,表示移除當前賬戶,因為我的pytorch賬戶之前存在。

之后需要安裝一些包,輸入Y,執行下一步。

之后就可以輸入箭頭所指的命令,激活環境:

就可以看到前面的base變為pytorch

之后我們要考慮怎么安裝pytorch。因為在接下來的步驟中會安裝cuda toolkit,所以不用單獨裝。

我們需要進入pytorch首頁:http://www.pytorch.org/

一般window使用conda,linux系統使用pip,我自己裝的10.1,你們根據自己電腦的顯卡,安裝合適的版本。

下圖是pytorch和顯卡驅動型號的適配圖:

前提要查看你的顯卡驅動是否安裝完成,如果可以看到下圖信息,表明你的顯卡驅動安裝好了。只要出現一個就ok。

 

conda install pytorch torchvision cudatoolkit=10.1

  記住要把最后 -c pytorch去掉,不然還是在原來的源下載,依舊很慢。

如果出現PackagesNotFoundError: The following packages are not available from current channels 這個錯誤,試試添加以下語句

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
# for legacy win-64
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/

這些就是需要下載的包,輸入y繼續。

如果依次輸入以下命令,輸出是True,那pytorch配置成功。

 現在你的pytorch環境就在你的anaconda文件下的envs文件內,envs里的文件的名字就是你命名的pytorch名字。在bin下可以找到python3.6。估計在配置pycharm項目環境是會用到。


免責聲明!

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



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