关于anaconda、cuda和cudnn的安装教程可以看我的这篇博客,安装ptorch和tensorflow的前期过程是一样的,这里只详细讲解pytorch的安装过程。


安装pytorch

1、打开anaconda prompt,创建新的虚拟环境,命令如下:

conda create -n pytorch-gpu-1.8.0 python=3.6

2、切换到新的虚拟环境下,并查看list:

acativate pytorch-gpu-1.8.0
conda list

3、下载torch

这里使用的方法是将torch下载到本地,然后进行安装,避免版本不一致等问题,

下载地址

注意:

cu开头的才是版本,

torch后面的1.8.0是torch的版本号,

cp后面的3.6表示python的版本

4、安装torch

在anaconda prompt中,在新创建的虚拟环境下(我这里是pytorch-gpu-1.8.0环境下),通过cd命令将目录切换到torch的下载目录下,

然后输入安装命令:

pip install torch-1.8.0-cp36-cp36m-win_amd64.whl(这里是你下载下来的文件名)

回车等待安装完成即可。

5、测试安装成功

在anaconda prompt中输入python进入python环境,依次输入:

import torch
torch.__version__  #1.8.0
torch.cuda.is_available()  #True

如果返回结果如注释所示,那就说明pytorch(gpu)安装成功,否在失败。

配置到pycharm上

打开pycharm编辑器,在File下选择Settings,点击最右边的设置,选择Add,然后选择现有环境,将目录定位到新创建的anaconda虚拟环境下的python.exe,一路点击确定退出,即可完成配置。

更多推荐