1. 程式人生 > >win10下使用anaconda安裝pytorch和torchvision

win10下使用anaconda安裝pytorch和torchvision

一、系統配置以及軟體版本

二、安裝過程

1、如果需要安裝GPU版本的,首先安裝CUDA,順序執行安裝程式即可,隨後安裝cudnn,將cudnn解壓,對應檔案複製到cuda安裝資料夾下即可。 2、安裝anaconda4.3,沒有什麼難度,選擇安裝自帶3.6版本的python。 3、開啟命令提示符(cmd),輸入:conda --version,出現anaconda版本資訊。輸入:conda info --envs,出現安裝的環境(初始時只有root環境,我的已經安裝了TensorFlow和pytorch)。 在這裡插入圖片描述 4、建立pytorch環境,輸入:conda create -n pytorch python=3.6,“-n”是“name”的縮寫,pytorch是你建立的環境的名稱,其實可以自己命名,後面是在環境裡安裝python,前面anaconda裡也裝了python,那是在root環境(主環境)安裝的。出現你將要安裝的包名和版本資訊,輸入y,繼續安裝。 在這裡插入圖片描述

5、安裝成功後,出現如下圖所示資訊,輸入:activate pytorch,進入pytorch環境裡面,這個時候pytorch環境相當於只是個空的框架,真正的pytorch還沒有裝上。(輸入:deactivate pytorch,退出pytorch環境) 在這裡插入圖片描述 6、如上圖所示,在pytorch環境中,輸入:conda install pytorch -c pytorch,開始安裝pytorch,同樣需要輸入y,繼續安裝。 7、安裝完成後,輸入:python,顯示python版本資訊,並進入到python裡,輸入:import torch,沒有任何提示則說明安裝成功。(我這裡在沒安裝torchversion的情況下,以為安裝了pytorch就可以,import torchvisio結果提示錯誤) 在這裡插入圖片描述
8、輸入:exit(),退出python,進入到pytorch環境,輸入:conda install torchvision -c pytorch,結果顯示沒有包。 在這裡插入圖片描述 9、那我們換一種安裝方式,輸入:pip install torchvision,會看到安裝資訊,最後出現“Successfully……”,說明安裝成功。然後我們重複步驟7,就顯示torch和torchvision都安裝成功。 在這裡插入圖片描述 以上就是安裝pytorch的全部過程,供大家參考。