Pytorch 01 Pytorch安装及环境配置

需要注意的是需要先安装好CUDA(安装如下),然后再根据相应CUDA版本选择Pytorch相应版本安装。

Pytorch官网https://pytorch.org/中查看相关版本及对应系统,在命令行中输入"Run this Command"中代码即可

Pytorch 01 Pytorch安装及环境配置

其中需要注意的是自己电脑的CUDA版本,CUDA版本必须对应,否则Pytorch会运行出错,其中CUDA安装过程如下:

首先需要查看自己电脑对应的CUDA版本,安装的CUDA版本如果与电脑不匹配也会出现运行错误。

打开NVIDIA控制面板(找不到在控制面板搜索即可),点击帮助->系统信息->组件->3D设置中即可查看CUDA版本

Pytorch 01 Pytorch安装及环境配置

如图笔者电脑系统对应的CUDA版本为CUDA10.

打开CUDA官网安装位置https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

Pytorch 01 Pytorch安装及环境配置 

选择相应版本,推荐选择本地[exe(local)]安装即可。