第一步:安装torch
(1)需要到torch官网上查看,符合自己电脑配置的torch版本。
选择自己电脑的系统,CUDA(可在控制面板,系统信息里查看,我的是9.1),官网上会给出对应版本的连接。
将此链接复制,留用。
注:CUDA最后一个选项是None,我就选的这个选项。因为我的CUDA是9.1,官网没有所以就选了None。
(2)打开Anaconda Prompt.
我的电脑是windows系统,所以输的是conda
首先,输入 conda env list 查看已有的环境。
然后,输入conda create-n py37_torch131 python=3.7,新建一个名为py37_torch131的虚拟环境。
然后,输入activate py37_torch131,激活这个环境。
激活环境后,输入pip install 此处加上复制我们前面复制的链接。
然后就开始安装啦。
注意:如果网络不好,可能会报readtime out错误,此时要找个好网络,或者换个源。
我是用手机给电脑开的热点,没办法家里网络太差了。
(3)如果选择使用jupyter进行编程的话,我们发现在新建python文件时依然只有python3这一个选项。
这是因为jupyter还没反应过来,这时候需要安装一个插件。安装完插件再刷新一个jupyter就好了。
安装插件的步骤如下:
首先,输入 conda env list 查看已有的环境。
然后,输入activate py37_torch131,激活这个环境。
然后,输入conda install ipykernel,如果这个不可行,可再输入pip install ipykernel。
(4)我们现在这个虚拟环境只安装了torch这一个包。
如果需要安装其他的包,在这个环境下,输入pip install 其他包的名字,就可以了.
比如说,使用pip install numpy,去安装numpy这个包。