win10安装cuda+Anaconda中的pytorch

由于笔者自己在下载pytorch的时候,踩过诸多坑。一直想着出一套完整的教程,但是时间久远,很多之前保存的下载过程图片都没有了。
特写此篇,来进行一下小小的参考。(第一次写,请见谅~(抱拳!))

一、win10安装CUDA+cuDNN

首先要查看自己电脑上的NVIDIA、CUDA信息
然后安装对应的CUDA(版本可小于等于电脑对应的信息)
英伟达官网下载链接:https://developer.nvidia.com/cuda-toolkit-archive
下载完CUDA之后,要安装与其对应版本的cuDNN(必须对应!!! )

二、Pytorch安装(划重点!!!)

接下来就是进行Pytorch的安装
我推荐是使用的是Anaconda进行pytorch进行安装下载,因为便于管理。
Anaconda详细安装及使用教程(带图文)链接:https://blog.csdn.net/ITLearnHall/article/details/81708148

pytorch安装https://blog.csdn.net/qq_45281807/article/details/121294644

但是!!!安装这种方法可能会一直显示false,因为清华源不支持pytorch了,如果用conda默认下载的,会出现cpu的版本。
这个时候,我们就需要手动下载安装包,进行配置咯~

手动下载配置:

不使用命令,离线下载之后安装即可(根据官网的下载命令查找对应版本号)CUDA,Python,torch,torchvision版本号一定要对!!
各种版本的torch,torchvision,torchaudio下载链接表单https://download.pytorch.org/whl/torch_stable.html
安装上述.whl文件

 pip install wheel
pip install 路径\文件名.whl

验证下载Pytorch是否成功

python
import torch
import torchvision
print(torch.cuda.is_available())

返回True表示CUDA配置成功~

参考教程:

  1. win10 安装Pytorch GPU版+CUDA+cuDNN(篇幅较长保姆教程) (含CUDA11.1与cuDNN8.0.4安装包)
  2. 解决torch.cuda.is_available()一直返回False的玄学方法之一
  3. 环境配置—Cuda和Cudnn版本对应关系
  4. Anaconda+cuda+cudnn+Pytorch全家桶安装
  5. Anaconda详细安装及使用教程(带图文)

猜你喜欢

转载自blog.csdn.net/m0_49960034/article/details/122910675