Windows11 配置Cuda cuDNN Pytorch环境

文章目录

  • 1 安装Cuda
      • Note 安装失败的情况
  • 2 安装cuDNN
  • 3 安装Anaconda
  • 4 安装Pytorch
  • 5 使用Pycharm进行验证
  • 6 结束

1 安装Cuda

  1. 进入 developer cuda: https://developer.nvidia.com/

Windows11 配置Cuda cuDNN Pytorch环境_第1张图片
Windows11 配置Cuda cuDNN Pytorch环境_第2张图片

Windows11 配置Cuda cuDNN Pytorch环境_第3张图片

  1. 之后自定义路径进行安装即可
  • 必装CUDA,其他情况自己选择
    Windows11 配置Cuda cuDNN Pytorch环境_第4张图片

  • 自定义路径 Document与Development路径

Windows11 配置Cuda cuDNN Pytorch环境_第5张图片

  • 等待安装成功即可

Windows11 配置Cuda cuDNN Pytorch环境_第6张图片

Note 安装失败的情况

原因:

  1. 没有 Visual Studio
    Cuda需要一些C++库的支持,因此需要安装Visual Studio

Windows11 配置Cuda cuDNN Pytorch环境_第7张图片

VS下载: https://visualstudio.microsoft.com/zh-hans
Windows11 配置Cuda cuDNN Pytorch环境_第8张图片

2 安装cuDNN

  1. 下载: https://developer.nvidia.com/
    Windows11 配置Cuda cuDNN Pytorch环境_第9张图片
  • 登录Nvidia账号之后下载

Windows11 配置Cuda cuDNN Pytorch环境_第10张图片

  • 下载后解压缩,将对应的文件复制到Cuda的对应文件夹中

  • 默认情况下 Cuda位于C->program Files->NVIDIA GPU Toolkit文件夹下

Windows11 配置Cuda cuDNN Pytorch环境_第11张图片

3 安装Anaconda

  1. 从清华镜像源下载Anaconda https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

Windows11 配置Cuda cuDNN Pytorch环境_第12张图片

  1. 自定义路径安装即可(默认不添加环境变量,我们自己添加)

  2. 添加环境变量,这里不提供基本的操作步骤,请自行搜索windows配置环境变量的方式

Windows11 配置Cuda cuDNN Pytorch环境_第13张图片

  1. 设置镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes 
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

4 安装Pytorch

  1. 进入Anaconda,创建新环境

Windows11 配置Cuda cuDNN Pytorch环境_第14张图片

  1. 进入环境,进入Prompt CMD命令行
    Windows11 配置Cuda cuDNN Pytorch环境_第15张图片

  2. 安装Pytorchhttps://pytorch.org/get-started/locally/, 根据版本进行选择,CUDA版本只能高不能低

Windows11 配置Cuda cuDNN Pytorch环境_第16张图片

Windows11 配置Cuda cuDNN Pytorch环境_第17张图片

5 使用Pycharm进行验证

  1. 添加解释器
    Windows11 配置Cuda cuDNN Pytorch环境_第18张图片

Windows11 配置Cuda cuDNN Pytorch环境_第19张图片

Windows11 配置Cuda cuDNN Pytorch环境_第20张图片

  1. 执行

Windows11 配置Cuda cuDNN Pytorch环境_第21张图片


6 结束

在安装的过程中踩了一个坑,就是我这台电脑有木马病毒,Cuda和Nvidia的驱动一直不能安装成功(已经安装VS,Cuda没有给提示),重置电脑之后就可以了,算是一点小曲折吧。

严重怀疑是病毒导致我安装失败!!!

你可能感兴趣的:(编程环境配置,windows,pytorch,python)