環境
我這裡用的是Anaconda最新的版本24.9.2,建立的pyton版本是3.9,虛擬環境名叫pytorch,下載的pytorch版本是12.1
初步配置
建立虛擬環境
首先進入Anaconda的終端,然後執行conda create -n xxx python=3.9
建立一個虛擬環境,這裡xxx是虛擬環境的名字,後面跟著的是python的版本
然後用conda activate xxx
啟用環境,退出環境用conda deactivate xxx
如果需要刪除環境,用conda rm -n xxx --all
刪除
下載pytorch
然後進入pytorch的官網,下滑找到自己需要的pytroch版本,然後複製出現的命令在conda對應的虛擬環境中進行下載 (漫長的等待...這裡由於網路問題會下載失敗,重試幾次即可,注意有顯示卡的不要到網上去換映象源,新版本的pytorch用映象源下載不成功,是CPU版本的)
下載完成後進入python環境,然後import torch
匯入torch庫,再用torch.cuda.isavailiable()
檢視是否正確安裝好pytorch,如果安裝成功,有顯示卡的出現true,沒有的出現false
配置pycharm
這裡我下載的是最新的pycharm,進入pycharm之後需要自己配置環境,新建一個專案需要獲取Anaconda的環境,如下所示:
如果沒有自動彈出,選擇自定義環境後選擇現有,型別選擇conda,然後路徑到conda的目錄中的Liabrary的bin的conda.bat,就可以自動配置環境了,要注意的是最下面的環境要選擇對應的虛擬環境!!!比如前面建立的是pytorch就選擇pytorch,不能預設!!!!,之後建立專案就可以了
同樣的,進入專案後在終端中import torch
匯入torch庫,再用torch.cuda.isavailiable()
檢視是否正確安裝好pytorch,如果安裝成功,有顯示卡的出現true,沒有的出現false
安裝Jupyter Notebook
在新版的pytorch和python中,直接用conda install nb_conda
是不行的,要用conda install nb_conda_kernels
下載,完成之後輸入jupyter notebook
就可以進入介面,然後建立一個新的jupyter book專案即可
同樣的,我們在book中import torch
匯入torch庫,再shift + enter
後,用torch.cuda.isavailiable()
,然後按下shift + enter
檢視是否正確安裝好pytorch,如果安裝成功,有顯示卡的出現true,沒有的出現false
這裡shift+enter
的作用是切換到下一個程式碼塊並且執行上一個程式碼塊,如果沒報錯說明成功