Win10 + CUDA12.3 + PyTorch 3.0 手动安装全攻略:从环境配置到验证

张开发
2026/5/5 17:04:09 15 分钟阅读
Win10 + CUDA12.3 + PyTorch 3.0 手动安装全攻略:从环境配置到验证
1. 环境准备搭建深度学习开发基础在Windows 10上配置CUDA 12.3和PyTorch 3.0之前我们需要先准备好基础环境。我遇到过不少新手直接跳过后面的步骤导致安装失败的情况所以这里会详细说明每个环节的注意事项。首先确认你的硬件配置是否支持CUDA。打开设备管理器查看显卡型号NVIDIA显卡需要是GTX 10系列及以上如RTX 3060、Tesla V100等。我实测过GTX 970也能运行但性能会打折扣。接着检查系统版本Win10需要是1903及以上建议更新到最新版。按下WinR输入winver就能看到具体版本号。Python环境推荐使用Anaconda管理它能完美解决不同项目间的依赖冲突问题。下载Anaconda3-2023.09版本自带Python 3.11安装时注意三个关键点安装路径不要有中文和空格我习惯用D:\Anaconda3勾选Add Anaconda to PATH选项虽然官方不推荐但实测更方便安装完成后在cmd执行conda init初始化环境验证安装成功的正确姿势是打开新终端窗口重要依次执行conda --version # 应显示conda 23.x python --version # 应显示Python 3.11.x where python # 检查路径是否在Anaconda目录下2. CUDA 12.3安装实战指南CUDA安装是个精细活我见过有人反复重装5次都没成功。首先到NVIDIA官网下载cuda_12.3.1_546.12_windows.exe注意要选local版本完整离线包约3GB。安装时有几个关键选择自定义安装而不是快速安装取消Visual Studio Integration除非你确实要用VS保留Driver components除非你确定驱动已最新安装过程中常见的坑是报错NSight安装失败这其实不影响CUDA使用直接忽略即可。安装完成后需要手动添加环境变量这是很多教程没讲清楚的地方CUDA_PATHC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.3 PATH中添加%CUDA_PATH%\bin;%CUDA_PATH%\libnvvp验证安装不要只用nvcc -V我推荐完整的检查流程nvcc --version # 应显示12.3 nvidia-smi # 查看驱动版本和GPU状态 cd %CUDA_PATH%\extras\demo_suite .\bandwidthTest.exe # 应显示Result PASS .\deviceQuery.exe # 应显示Detected 1 CUDA Capable device3. PyTorch 3.0手动安装技巧官方推荐的conda安装方式在国内经常超时我来分享实测可用的手动安装方案。首先在清华镜像站下载这些组件注意版本匹配pytorch-3.0.0-py3.11_cuda12.3_cudnn8_0.tar.bz2torchvision-0.15.1-py311_cu123.tar.bz2torchaudio-2.0.1-py311_cu123.tar.bz2cudatoolkit-12.3.1-h774522f_1.conda下载完成后按这个顺序安装管理员权限运行cmdconda install --use-local cudatoolkit-12.3.1-h774522f_1.conda conda install --use-local pytorch-3.0.0-py3.11_cuda12.3_cudnn8_0.tar.bz2 conda install --use-local torchvision-0.15.1-py311_cu123.tar.bz2 conda install --use-local torchaudio-2.0.1-py311_cu123.tar.bz2安装后别急着验证先配置conda国内源加速后续依赖安装conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge conda config --set show_channel_urls yes4. 完整环境验证与问题排查验证PyTorch是否正常工作的正确姿势是运行以下测试代码import torch print(torch.__version__) # 应显示3.0.0 print(torch.cuda.is_available()) # 必须返回True x torch.rand(5,3).cuda() # 创建张量并转移到GPU print(x.device) # 应显示cuda:0常见问题解决方案如果torch.cuda.is_available()返回False检查CUDA和PyTorch版本是否匹配运行python -m torch.utils.collect_env查看环境详情更新显卡驱动到最新版遇到Could not load dynamic library cudnn64_8.dll错误从NVIDIA官网单独下载cuDNN 8.9.4将bin目录下的dll文件复制到CUDA的bin目录内存不足报错torch.cuda.empty_cache() # 清空缓存 torch.set_default_tensor_type(torch.FloatTensor) # 改用CPU模式最后在PyCharm中配置Conda环境时要注意选择Existing environment路径指向Anaconda安装目录下的python.exe不要选虚拟环境路径。我建议新建一个test.py运行上述验证代码确保IDE环境配置正确。

更多文章