pytorch安装详细步骤

环境配置步骤

如果显卡支持GPU，在命令行处，输入nvidia-smi ，查看自己驱动的版本。
参考文章

官网选择型号

官网

清华源快速安装

参考博文

安装完检验

输入torch.cuda.is_available()，显示True则意味着可以使用GPU。

pycharm设置项目环境

这一步是容易忽略导致操作错误的地方，设置环境项目注意如果是导入已有的虚拟环境，需要选择conda environment，如果是新建虚拟环境，则选择virtualenv environment。
这里因为需要导入已有的安装pytorch的环境，所以正确操作如下:

之后可以看到对应的编译器已经添加到：

项目显示编译器未设置，进行再次设置。

查看pytorch版本号

import torch
print(torch.__version__)

更新pytorch

在官网选择需要更新的版本号，会出现对应的下载的代码。