1. 程式人生 > 其它 >安裝Anaconda和配置Pytorch虛擬環境

安裝Anaconda和配置Pytorch虛擬環境

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就不會出錯了