最近跑深度学习项目,不同的代码有时需要不同的pytorch版本,因此经常需要创建新的虚拟环境并安装pytorch,记录一下最快速的方法(对本人来说)
1.anaconda创建新的虚拟环境
为便于管理,我创建的虚拟环境都是python 3.7
首先,在linux下,使用vim ~/.condarc
修改配置文件,将以下国内源拷贝进去,这样新建虚拟环境就嘎嘎快:
channels:
2 - https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
3 - https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
4 - https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
5 - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
6 - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
7 - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
8 - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
9 ssl_verify: true
然后用以下命令新建虚拟环境,很快就创建好了:
conda create -n env_name python=x.x
2.安装需要的pytorch版本:
在该链接https://pytorch.org/get-started/previous-versions/选择合适的pytorch版本,首先使用pip命令进行安装(我用conda总是会出现网络问题,从没成功过),如果pip也报错,就进入安装命令里面的网址,手动下载合适的whl安装包,然后pip install xxx.whl安装包即可。