安裝Anaconda和配置Pytorch虛擬環境
阿新 • • 發佈:2021-09-08
Linux伺服器配置Pytorch虛擬環境
一、安裝Anaconda
1. 下載Anaconda檔案
2. bash命令安裝
bash Anaconda3-5.2.0-Linux-x86_64.sh
3. 驗證是否安裝成功
在終端輸入anaconda,如果出現如下圖所示命令,則代表安裝成功
4. 如果顯示無法找到命令怎麼辦?
開啟配置檔案
vim ~/.bashrc
在檔案最後新增語句
export PATH=/home/XXX/anaconda3/bin:$PATH #XXX為使用者名稱
最後更新配置檔案即可
source ~/.bashrc
二、配置Pytorch虛擬環境
1. 建立虛擬環境,環境名為pytorch
conda create -n pytorch python=3.6
source activate pytorch
2. 查詢對應版本的pytorch並安裝
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch #網路不好的時候刪除後面的 -c pytorch 就不會用國外的源了,有效 #升級到特定版本的torch pip install --upgrade torch==1.8.0
3. 安裝完畢測試
import torch import torchvision print(torch.__version__) #輸出PyTorch版本
4. ipython測試出錯:沒有指定的包torch
原因:ipython與該新建環境指定的python路徑不一致
解決方法:修改修改IPython的配置檔案,將python的地址改為環境中的python地址
#進入環境,檢視環境中python的路徑 which python #/home/zhengwenqing/anaconda3/envs/pytorch/bin/python #檢視ipython的路徑 which ipython # /home/zhengwenqing/anaconda3/bin/ipython #修改IPython的配置檔案第一行 vim /home/zhengwenqing/anaconda3/bin/ipython #(原本)#!/home/zhengwenqing/anaconda3/bin/python#(修改為環境中的python路徑) #!/home/zhengwenqing/anaconda3/envs/pytorch/bin/python #進入ipython,報錯沒有ipython module,用pip安裝即可 pip install IPython #然後ipython就是環境對應的版本了,這時候import torch就不會出錯了