最近跑实验有用到pytorch
我的anaconda是已经安装好了,也安装了tensorflow
现在用conda命令安装一下pytorch
首先创建虚拟环境
为pytorch创建一个虚拟环境,这是推荐的做法,这样在已有其他环境如tensorflow时,就不会互相冲突。在终端输入:
conda create -n pytorch python=3.5
这就创建了一个名为pytorch,python版本为3.5的虚拟环境。于是我们可以把Pytorch安装在这个环境下面。
首先激活这个环境:source activate pytorch
source activate pytorch
在Pytorch的官网,根据自己的机器配置选择对应的安装命令。
我的是
在终端输入:
conda install pytorch torchvision cudatoolkit=8.0 -c pytorch
可能会有一些需要升级或安装的包,一路升级安装就行了。