Link Search Menu Expand Document
Table of contents
  1. CUDA+cuDNN安装
  2. PyTorch安装

CUDA+cuDNN安装

下载的CUDA版本必须要与所支持的驱动相对应

可以先将显卡更新至最新版本后再下载。下载安装NVIDIA显卡驱动
NVIDIA官网:https://www.nvidia.cn/geforce/drivers/
下载驱动程序,注册账号。
NAVIDIA.png
安装NVIDIA完成后,打开GEFORCE Experience,打开驱动程序,检查更新文件,若有,则下载更新文件
![](NVIDIA更新.png
更新后可进入cmd,输入nvidia-smi查看CUDA Version以及 Driver Version
Version查看.png
完成之后根据上述两个参数来判断下载哪个版本的CUDA和cuDNN,下载时要注意一定要让cuDNN与CUDA的版本号完全一样
CUDA下载网址:https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64
cuDNN下载网址:https://developer.nvidia.com/rdp/cudnn-archive
cuda版本对应.png
解压cuDNN,解压下载CUDA文件夹,选择自定义安装,下一步,勾选全部选项
cuda1.png
cuda2.png
安装完成之后,找到根目录C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7
将cuDnn文件夹中的bin、include、lib文件夹中的内容复制到cuda中的同名文件夹中,结果如:
cuDNN复制.png
以上操作完成CUDA+cuDnn安装,打开cmd,输入nvcc -V查看cuda信息:
cuda完成.png

PyTorch安装

完成显卡配置后,安装PyTorch
打开Anaconda prompt,输入python -V查看python安装版本
python版本.png
输入conda create -n pytorch python=3.9
创建环境.png
其中pytorch是创建环境变量名,python=3.9是当前环境的版本数
创建环境后,需加载某些包,同意输入y
package load.png
等待包下载完成后,输入activate pytorch,若左侧从base->pytorch,则成功进入pytorch环境
pytorch完成.png
然后输入conda install pytorch
等待安装完成后验证安装是否有效:

  • 首先输入conda activate pytorch进入pytorch虚拟环境
  • 输入python,进入python界面
  • 输入 import torch
  • 输入 torch.cuda.is_available()
    若输出结果为Ture,则安装成功,若为False,则失败
    验证torch.png
    若输出为False 输入pip list,查看当前环境下有哪些包,若没有torch,则进入其官网进行下载
    pytorch官网:https://pytorch.org/
    torch.png
    进入官网下拉,找到对应torch版本
    torch load.png
    输入conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge
    torch load2.png
    下载完成后,再一次输入pip list 查看是否存在torch包
    若存在,再一次检测安装是否成功(输出为Ture)


    若还是错误,尝试直接下载 .whl文件进行
    下载地址:https://download.pytorch.org/whl/torch/
    选择匹配自己python版本的torch、torchaudio和torchvision,cu115代表匹配cuda11.5版本,cp39表示匹配python3.9版本,win就表示windows
    torch下载:https://download.pytorch.org/whl/torch/
    torchaudio下载:https://download.pytorch.org/whl/torchaudio/
    torchvision下载:https://download.pytorch.org/whl/torchvision/

    三个文件下载后,放在平时安装的位置:C:\Users\lenovo,其后在cmd中将三个文件按顺序安装
    cmd输入:pip install 文件名
    安装成功后,文件夹中出现一下六个文件

    再重新进入python环境,检查是否成功安装
    验证torch.png