安装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 保存 ,如果不需要了,注释掉就好