需要注意的是需要先安装好CUDA(安装如下),然后再根据相应CUDA版本选择Pytorch相应版本安装。
Pytorch官网https://pytorch.org/中查看相关版本及对应系统,在命令行中输入"Run this Command"中代码即可
其中需要注意的是自己电脑的CUDA版本,CUDA版本必须对应,否则Pytorch会运行出错,其中CUDA安装过程如下:
首先需要查看自己电脑对应的CUDA版本,安装的CUDA版本如果与电脑不匹配也会出现运行错误。
打开NVIDIA控制面板(找不到在控制面板搜索即可),点击帮助->系统信息->组件->3D设置中即可查看CUDA版本
如图笔者电脑系统对应的CUDA版本为CUDA10.
打开CUDA官网安装位置https://developer.nvidi