PyTorch 環境配置及安裝
1. 安裝 anaconda
2. 為 PyTorch 創建單獨的環境,便於管理。
-
打開 Anaconda Prompt
-
為 PyTorch 創建單獨的環境
python 版本與 anaconda 中 python 版本相同即可。
conda create -n pytorch python=3.8
輸入 y 開始安裝需要的包並創建環境(如果出現了 ERROR 並且你此時正在使用代理,嘗試關閉代理)。
創建成功。 -
激活環境並查看環境下的包
conda activate pytorch
pip list
-
如果沒有 pytorch 包,需要下載安裝。
3. PyTorch 安裝
-
pytorch 首頁:https://pytorch.org/
選擇相關參數,如果電腦沒有Nvidia顯卡或者只有集顯,CUDA 選擇 CPU 即可。然后運行下面的指令。 -
在 pytorch 環境下運行指令。
-
安裝檢驗
命令行輸入 python 進入 python 環境,執行import torch
,如果沒有報錯,說明安裝成功。
輸入torch.cuda.is_available()
檢查 GPU 是否可用, 返回 true 即表示可用。
至此,PyTorch 的安裝配置完成。
4. 在新建的 pytorch 環境中安裝 Jupyter
-
進入 pytorch 環境
conda activate pytorch
-
安裝需要的包
conda install nb_conda
-
屏幕上顯示 done 即安裝成功。
-
使用 Jupyter
jupyter notebook