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
版权声明：本文为weixin_74986983原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。