深度学习三件套安装依赖指南

安装依赖指南

标准的深度学习三件套包括pytorch, cuda, cudnn。根据经验,一般安装顺序如下:

  1. 用conda安装cudaconda install cudatoolkit==11.7,使用清华源(到官网确认安装了符合pytorch要求的cuda版本)
  2. 用pip安装pytorch,使用清华源。
  3. 安装合适版本的cudnn。

如何安装cuda

用conda安装即可:conda install cudatoolkit==11.7,conda可以使用国内清华源。

如何下载pytorch gpu离线版本

参考博客 https://blog.csdn.net/qq_51206550/article/details/127835129

下载pytorch gpu版本太慢,到官方镜像 https://download.pytorch.org/whl/torch/,下载需要的GPU版本,比如torch-1.13.1+cu117-cp39-cp39-win_amd64.whl,其中cu117代表cuda版本为11.7,cp37代表python版本为3.7。

https://blog.csdn.net/qq_41710226/article/details/125548986

如何安装cudnn

在windows安装cudnn详见"2. Installing cuDNN on Windows"一章,简单讲就是下载cudnn的压缩包,放到某个路径,再将bin目录添加到path。

你可能感兴趣的:(踩坑笔记,pytorch,python,深度学习)