PyTorch[GPU版] 安装教程

1. 查看CUDA版本

打开cmd，输入nvidia-smi。若没有CUDA，则会报错，可自己安装CUDA

 如图所示，该CUDA版本为11.1

2. 下载pytorch

进入pytorch官网，选择合适的pytorch下载，pytorch的CUDA版本要比电脑的CUDA版本低

 CUDA11.7比电脑的11.1要大，所以选择其他的版本。选择好后将红色框框的内容复制，而后进行下载（可用pip下载，pycharm可打开终端Terminal进行下载）

 粘贴后即可下载

3. 检查

导入模块

import torch

若不报错，则已有pytorch

查看pytoch版本

print(torch.__version__)

若打印结果有cu，则版本为GPU；若有cpu，则版本为CPU