1 查看显卡驱动
根据图中给出的CUDA版本进行对应版本安装
安装CUDA11.4
直接进入CUDA11.4官网进行下载
cuda下载后,进行安装。安装步骤可参考该文
CUDA环境配置
环境配置可参考该文
安装完成后,在命令窗口输入:ncvv -V。查看是否安装正确。
cuDNN下载及安装
cuDNN下载官网
安装步骤如该文
安装后,检查是否安装成功。
- 首先执行:deviceQuery.exe,查看是否出现如下界面:
- 然后执行bandwidthTest.exe,出现如下界面,则代代表cuDNN也安装成功。.
创建虚拟环境
这里安装的是python=3.7的版本。
安装完之后,要激活该版本
安装Pytorch
进入Pytorch官网
Pytorch官网以前版本
注意:CUDA11.4可以安装CUDA11.3版本对应的pytorch。
最后,检测pytorch是否安装成功,输入python,再import torch,正常导入即成功安装。