2020-11-17

pytorch课程之关于pytorch环境创建

创建虚拟环境

进入到某一虚拟环境之中

查看该虚拟环境的python版本并退出

在该虚拟环境下安装工具包

CUDA安装

1.可进入pytorch官网查看各版本pytorch所对应的CUDA版本。
2.搜索cuda，进入cuda官网→cuda安装

安装完后，查看cuda版本

如图，cuda版本为10.1。证明安装cuda10.1成功。

CUDNN安装

进入官网→登录→选择已安装cuda对应的cudnn下载→
将下载好的cudnn解压，将里面三个文件夹里的文件分别复制到“C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1”对应文件夹中→验证CUDNN是否安装成功：

出现一个错误：
是由于显卡配置低造成的，换成cuda9.2版本：

出现Pass表示cudnn安装成功