1.pytorch cpu版本网站: https://pytorch.org/ gpu版本网站: https://pytorch.org/get-started/previous-versions/ gpu版本可能下载太慢,去清华同方 https://blog.csdn.net/zengding5524/article/details/109484757 2.测试是否成功代码 import torch # 如果pytorch安装成功即可导入 print(torch.cuda.is_available()) # 查看CUDA是否可用 print(torch.cuda.device_count()) # 查看可用的CUDA数量 print(torch.version.cuda) # 查看CUDA的版本号
www.bilibili.com