windows安装CUDA

说明

• 适用于你的电脑为nvidia的显卡

第一步

• 1. 先运行nvcc -V 查看cuda版本，不过应该都是未识别的命令，如下图所示
     
• 2. 打开pytorch的官网 https://pytorch.org/get-started/locally/ ，查看其支持最新的cuda为多少版本，例如：当前支持到cuda11.8

第二步

• 1. 下载cuda：传送门https://developer.nvidia.com/cuda-toolkit-archive
• 2. 选择版本11.8，如果后面pytorch支持更高的cuda，可以安装新版的cuda
     

选择win10或者win11

• 3.等待下载完成，双击cuda_11.8.0_522.06_windows.exe运行，直接下一步。。。

第三步

• 测试是否正确安装，需要新打开一个终端，输入nvcc -V 如图
  

最后

• 再安装带cuda版本的pytorch: pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

• 这一步应该可以在安装cuda之前安装，本人没有测试过，只实验了先安装cuda，再安装pytorch