文章目录

  • 下载Anaconda3
    • 安装,看着点next就行
    • 比较懒所以自动添加path
    • 测试
  • cuda
    • 安装的时候不能改路径
    • 如果出现报错,关闭杀毒软件
    • 一直下一步就好
    • 取消勾选“CUDA”中的“Visual Studio Intergration”
    • 一直下一步即可
    • 测试
    • 安装成功
  • cudnn
    • 解压后将这三个文件夹复制到CUDA的安装目录中
    • 没选择自动添加环境变量的话需要手动设置
    • 验证
    • 安装成功
  • pytorch
    • 如果没有你的cuda版本
    • 下滑,找到适合你的版本即可

下载Anaconda3

官网

安装,看着点next就行

安装的时候注意你自定义的文件夹必须是空的

比较懒所以自动添加path

测试

win+r,输入cmd,打开终端
输入conda info,就能看到你的conda版本以及自带的python版本

输入activate就能进入到默认的base环境

cuda

命令行输入nvidia-smi

根据自己的版本下载

安装的时候不能改路径

如果出现报错,关闭杀毒软件

一直下一步就好

取消勾选“CUDA”中的“Visual Studio Intergration”

一直下一步即可

测试

nvcc –version # 检查CUDA的版本号
set cuda # 查看CUDA设置的环境变量

安装成功

cudnn

先注册后下载

解压后将这三个文件夹复制到CUDA的安装目录中


没选择自动添加环境变量的话需要手动设置

验证

win+R ,cmd打开,切换目录到demo_suit
以下是默认路径,你只用改一下版本号v12.1
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1\extras\demo_suite
输入
bandwidthTest.exe

deviceQuery.exe

安装成功

pytorch

官网下载

如果没有你的cuda版本

下滑,找到适合你的版本即可