windows10+cuda+pytorch

仅作为个人cuda安装记录。

1、安装NVIDIA 驱动程序；

2、安装VS2019；

3、安装Cuda  :  cuda_10.0.130_411.31_win10.exe

4、自定义安装cuda, 取消掉VS的选项；

5、环境变量

6、安装cuda 10 对应的pytorch版本；

pip install torch==1.2.0 torchvision==0.4.0 -f https://download.pytorch.org/whl/torch_stable.html

7、装完之后检查是否成功；

 

2020-4-13 补--------------

 

8 将下载的cudnn解压

 

9 把cudnn目录下的文件覆盖至cuda目录下；

 

完-------