前言

仅记录学习过程，有问题欢迎讨论

因为网上博客参差不齐，我也踩了很多坑，留下我自己成功的经验哈哈。

1.安装CUDA和CUDNN
参考这个博客：

https://blog.csdn.net/shdabai/article/details/131248257

2.安装gpu版本 pytorch
参考这个博客：

https://blog.csdn.net/m0_62627216/article/details/135473628?utm_medium=distribute.pc_relevant.none-task-blog-2_defaultbaidujs_baidulandingword~default-1-135473628-blog-131798391.235^v43pc_blog_bottom_relevance_base1&spm=1001.2101.3001.4242.2&utm_relevant_index=4

3.测试

import torch
print(torch.__version__)
print(torch.cuda.is_available())

true 就成功啦~