安装Anaconda和配置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就不会出错了