1.安装cuda
安装cuda之前应该查看自己电脑适合什么版本可以先将驱动升级(这一步很重要,比如驱动精灵之类的驱动更新软件)
电脑右键→NVIDIA控制面板→系统信息→组件:可以看到下面的NVCUDA这里面显示是你可以装的最高版本
在这里我们已经知道我们的驱动适应的最高版本的cuda,因此下一步就是去官网下载对应版本的cuda:https://developer.nvidia.com/cuda-toolkit-archive
我们的最高版本是10.2,但在这里我选择安装cuda10.1,下载完成之后直接安装,里面安装的过程默认所选就行。
查看我们安装的cuda版本:cmd进入命令提示符中输入nvcc --version
2.安装cudnn
cudnn其实就是cuda的一个组件,在NVIDIA官网上找到就行
cuda+cudnn+anaconda+pycharm(安装pytorch)
最新推荐文章于 2024-08-19 16:45:16 发布
本文详细介绍了如何在Windows环境下安装CUDA、CUDNN,以及如何通过Anaconda创建环境并安装PyTorch。首先确保驱动升级,然后从NVIDIA官网下载并安装CUDA,接着安装CUDNN并将其放入CUDA安装目录。验证CUDA+CUDNN环境后,通过Anaconda安装conda,并创建PyTorch虚拟环境。最后,从PyTorch官网获取对应版本的安装命令,确保GPU支持的PyTorch成功安装。
摘要由CSDN通过智能技术生成