win10 cuda11.8 和 torch2.0 安装指南

俞绚义

于 2024-09-20 20:02:57 发布

阅读量691

点赞数 21

本文链接：https://blog.csdn.net/gitblog_06523/article/details/142396231

版权

本仓库提供了详细的步骤和说明文档，帮助您在Windows 10操作系统环境下安装CUDA 11.8和PyTorch 2.0。这一组合特别适用于那些需要在GPU环境下进行深度学习开发的用户。下面的步骤基于博主“小崔的技术博客”在CSDN上的文章指导，确保您的系统配置正确无误。

访问NVIDIA cuDNN官网，根据CUDA 11.8版本下载对应的cuDNN库。
解压下载的文件，并将bin, include, lib目录下的文件复制到CUDA的安装目录下的相应位置，例如C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8。
更新PATH环境变量，包括cuDNN的相关目录。

使用pip安装：在确保CUDA和cuDNN已正确配置的环境中，运行以下pip命令来安装PyTorch 2.0与相应的torchvision和torchaudio：
```
pip3 install numpy --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118
```
验证安装：安装完成后，在Python环境下，可以通过导入torch并检查.is_cuda属性来验证GPU支持是否正常工作。

通过遵循以上步骤，您应该能够在Windows 10平台上成功设置CUDA 11.8和PyTorch 2.0环境，为深度学习项目奠定坚实的基础。记得在实际操作中细心对照每一步骤，确保每个环节都顺利完成。