安装cuda 11 ,cudnn，pytorch 1.7

最新推荐文章于 2024-08-15 19:05:31 发布

jiuhu7

最新推荐文章于 2024-08-15 19:05:31 发布

阅读量3.9k

点赞数 4

分类专栏：深度学习相关文章标签： cuda cudnn pytorch

本文链接：https://blog.csdn.net/jiuhu7/article/details/110003831

版权

深度学习相关专栏收录该内容

5 篇文章 0 订阅

订阅专栏

本文详细记录了CUDA、CUDNN和PyTorch的安装步骤。首先检查电脑显卡驱动，然后从NVIDIA官网下载并安装CUDA11。接着下载CUDNN，并将其解压后的文件复制到CUDA安装目录。通过运行特定测试程序验证安装成功。最后，从PyTorch官网选择合适的版本下载，并提供了使用wheel文件通过pip安装的建议，以及torchvision和torchaudio的安装方式。

摘要由CSDN通过智能技术生成

安装cuda ,cudann，pytorch

1. 安装cuda

先康康电脑里面有没有
nvcc -V
结果当然是没有。
要安装首选需要康康自己的显卡能支持到多少的cuda
在这里插入图片描述
打开这个（可以通过搜索直接搜到。）

在这里插入图片描述
点击系统信息

在组件栏可以发现我的电脑支持11.0.228以下的driver。
想用新一点的，所以打算就直接安装cuda11
https://developer.nvidia.com/cuda-toolkit
去官网下载
进去发现直接就是11update。
找了一下才找到以前的版本下载的位置。
在这里插入图片描述

这个就是我想要的版本
挺大的，还好我的网速还不错。
下载完毕直接安装就可以了。我下载在C盘里面。

2.cudnn下载

https://developer.nvidia.com/rdp/cudnn-download
去这里下载
需要登录
可以直接微信登录就完事了，比较方便。
在这里插入图片描述
很明确的，就是下载这个
下载好是个压缩包，解压后有这些东西

然后进入 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0
将 bin include lib 都复制到这里面

验证是否安装成功 cuda cudnn

打开cmd
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\extras\demo_suite
运行 bandwidthTest.exe
在这里插入图片描述
运行 deviceQuery.exe

有PASS就是OK的。

安装pytorch 1.7

https://pytorch.org/
直接去官网
在这里插入图片描述
选好后下载好就行了。

参考：
https://blog.csdn.net/qq_18620653/article/details/105329219

时间 2021年5月15日
网络不好只能采取wheel的方式下载
https://pytorch.org/get-started/previous-versions/
https://download.pytorch.org/whl/torch_stable.html
去上面的第二个网址通过浏览器下载需要的wheel会比用pip下载稳定点，至于要下载哪个就只有参考第一个网址，或者通过pip返回出来的那个名字找就行。
下载好之后，通过

pip install torch-1.8.0+cu111-cp38-cp38-win_amd64.whl

安装就行了
我这次下载的是1.8版本的
链接：https://pan.baidu.com/s/1XBH5Zn5uPteURANU23taKA
提取码：mn3g
就顺带上传到网盘里了。
不过配套的 torchvision0.9.0+cu111 torchaudio0.8.0 就自己通过pip单独下载了。