Win10+GPU版Pytorch1.1安装

安装cuda

更新nvidia驱动

打开GeForce Game Ready Driver或在 GeForce Experience中下载符合自己gpu的程序。

 

 

 

选择cuda

1. 打开nvidia控制面板

    

点击帮助、点击系统信息、在点击组件在3D设置中可以看到cuda信息

 

 在我升级过nvidia驱动后，cuda的版本更新到了10.1。接下来下载cuda .  

cuda10.1安装完毕。

 

 

安装cuDNN

在安装了cuda10.1后选择对应的cuDNN版本v7.6.1

解压文件，然后添加bin目录到环境变量

 

安装cuDNN

在安装了cuda10.1后选择对应的cuDNN版本v7.6.1

解压文件，然后添加bin目录到环境变量

 

 

 

 

 

使用Anaconda prompt创建环境

conda create -n pytorch_gpu pip python=3.7

View Code