Win10+GTX1660Ti+CUDA10.1+cuDNN v7.6.4+Anaconda+PyCharm配置GPU版本PyTorch超详细步骤

最新推荐文章于 2022-04-16 21:28:24 发布

智信仁勇严道

最新推荐文章于 2022-04-16 21:28:24 发布

阅读量5.3k

点赞数 15

分类专栏： PyTorch 文章标签： cuda pytorch 深度学习 pycharm anaconda

本文链接：https://blog.youkuaiyun.com/qq_18620653/article/details/105375891

版权

PyTorch 专栏收录该内容

6 篇文章

订阅专栏

本文详述了PyTorch的配置流程，包括确认显卡能力、配置显卡驱动、CUDA、cuDNN，以及Anaconda和PyCharm的设置。提供在线与离线安装PyTorch的方法，最后验证GPU版本的PyTorch是否正确安装。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

超详细的步骤（细品你会有意想不到的收获）

前言：虽然配置PyTorch踩的坑要比caffe少的多，但在配置过程中还是遇到了些问题（比如如何选择显卡驱动、CUDA、cuDNN版本，三者之间依赖关系，为社么使用Anaconda、PyCharm等），网上涉猎的资料大部分只是“授人以鱼”。虽然费尽周折最终配置成功，但仍然稀里糊涂，所以本着“授人以鱼不如授人以渔”的原则，参考官方文档详细写下了PyTorch的配置过程，希望能够帮助到和我一样困惑的童鞋。本人能力有限，如有不妥之处还希望谅解。
配置过程中所有软件：
软件链接：https://pan.baidu.com/s/1bMbPR4cWORZ8gXlcD8PfcQ
提取码：5336
PyTorch包链接：https://pan.baidu.com/s/1QXnH5lV60aji7tmKl41ZMg
提取码：9zay
1、首先要确认自己电脑有张有能力跑深度学习的显卡。
2、配置显卡驱动、CUDA、cuDNN以及说明三者之间的关系（链接中有我详细介绍）https://blog.youkuaiyun.com/qq_18620653/article/details/105329219
3、Anaconda配置https://blog.youkuaiyun.com/qq_18620653/article/details/105335481
4、PyCharm配置以及关联Anacondahttps://blog.youkuaiyun.com/qq_18620653/article/details/105373487
5、以上步骤如果顺利完成，接下来安装PyTorch
PyTorch官网下载链接https://pytorch.org/
在这里插入图片描述
快捷键 Windows + R输入cmd回车

（1）在线安装方式

将上面的 pip install torch=1.4.0 torchvision=0.5.0 -f https://download.pytorch.org/whl/torch_stable.html 复制到命令框，然后回车静静等待安装。
在这里插入图片描述
出现Successful installed torch=1.4.0 torchvision=0.5.0说明PyTorch安装成功

（2）离线安装方式

https://blog.youkuaiyun.com/qq_18620653/article/details/105420158

测试GPU版本PyTorch

依次输入python、import torch、import torchvision、torch.version print(torch.cuda.is_available())如果不报错，并且返回torch版本，以及True。恭喜你PyTorch GPU版本安装成功。
在这里插入图片描述
至此，PyTorch配置过程到此结束，希望能够帮助到你。