windows10+cuda+pytorch

仅作为个人cuda安装记录。

1、安装NVIDIA 驱动程序;

2、安装VS2019;

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

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

windows10+cuda+pytorch_第1张图片

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、装完之后检查是否成功;

windows10+cuda+pytorch_第2张图片

 

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

 

8 将下载的cudnn解压

windows10+cuda+pytorch_第3张图片

 

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

 

完-------

你可能感兴趣的:(python,cuda)