pytorch安装总结

要考虑的因素有：os + python版本 + cuda版本 + 显卡驱动版本 + cudnn版本

一、cuda

cuda下载地址：https://developer.nvidia.com/cuda-toolkit-archive

查看cuda版本：

打开cmd，输入nvidia-smi 

 进入官网，选择对应版本，我选择的是11.6

配置好对应参数，download即可

除个别配置需要调整外，一路默认

 

 

 

查看系统变量，发现cuda变量已自动添加 

测试是否安装成功，输入nvcc -V，显示如下信息，则已安装

 二、cudnn

cudnn 下载地址： （需要注册登录才能下） https://developer.nvidia.com/rdp/cudnn-archive

 

将压缩包解压， bin/include/lib目录的内容全部拷贝到CUDA对应的目录下 

 编辑系统变量

 测试是否安装成功

①方式一

 ②方式二

 

 两个程序运行结果都为pass则表示cudnn安装成功

 三、pytorch

官网下载速度较慢，故采用下载离线包，然后pip安装的方式，whl包地址：https://download.pytorch.org/whl/torch_stable.html

选择适合本机的pytorch版本

 进入虚拟环境，pip install对应whl文件，然后测试安装成功与否