1200字范文,内容丰富有趣,写作的好帮手!
1200字范文 > Anaconda创建虚拟环境及pytorch安装

Anaconda创建虚拟环境及pytorch安装

时间:2020-02-27 19:03:41

相关推荐

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官网根据所需版本和设备情况选择:

因为我的cuda版本为10.1,所以点击图中箭头所指地方查看先前版本。对应得到安装命令:

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1

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

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。