查看已有虚拟环境
- 两者均可
conda info -e
conda env list
创建新的虚拟环境
conda create -n your_env_name python=x.x
以3.7为例(环境名字自取)
conda create -n py37 python=3.7
有时候会崩出这种错误conda install failed: conda.core.subdir_data.Response304ContentUnchanged,可以清理一下cache后重新创建(如遇其他错误,根据错误搜索查阅就好):
conda clean -i
显卡版本查看
我所使用的服务器是nvidia的显卡,直接调用命令即可。这里我们可以看到CUDA版本为10.1。
nvidia-smi
torch 安装
- 首先激活之前创建的额虚拟环境
source activate py37
- 查找对应版本的安装命令并安装
可以在Pytorch官网根据所需版本和设备情况选择:
因为我的cuda版本为10.1,所以点击图中箭头所指地方查看先前版本。对应得到安装命令:
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1
torch安装中可能遇到的一些问题,解决方法可见我之前的博客Anaconda安装torch中遇到的一些问题