windows 安装nvidaia驱动和cuda

安装nvidaia驱动和cuda

官网搜索下载驱动
https://www.nvidia.cn/drivers/lookup/

这里查出来的都是最高支持什么版本的cuda

安装时候都默认精简就行

官网下载所需版本的cuda包
https://developer.nvidia.com/cuda-toolkit-archive

安装成功但是nvcc -V 失败 ，除了安装时候默认的加入的环境变量外。

添加环境变量
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\CUPTI\libx64

安装pytorch

网站里找版本
https://pytorch.org/get-started/previous-versions/

pip install torch1.9.0+cu111 torchvision0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

pip 要是下载速度慢就切换清华源

mkdir %APPDATA%\pip
notepad %APPDATA%\pip\pip.ini

编辑文本

[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple

ctrl +s 保存 ，如果不需要了，注释掉就好