Anaconda Pytorch 深度學習入門記錄

cyMessi發表於2024-10-27

環境

我這裡用的是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的作用是切換到下一個程式碼塊並且執行上一個程式碼塊,如果沒報錯說明成功

相關文章