Anaconda创建虚拟环境及pytorch安装

文章目录

    • 查看已有虚拟环境
    • 创建新的虚拟环境
    • 显卡版本查看
    • torch 安装

查看已有虚拟环境

  • 两者均可
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官网根据所需版本和设备情况选择:
    Anaconda创建虚拟环境及pytorch安装_第1张图片
    因为我的cuda版本为10.1,所以点击图中箭头所指地方查看先前版本。对应得到安装命令:
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1

torch安装中可能遇到的一些问题,解决方法可见我之前的博客Anaconda安装torch中遇到的一些问题

你可能感兴趣的:(deep,learning,pytorch,深度学习,python)