深度学习环境配置【anaconda+CUDA+cudnn+pytorch-gpu】

在配置之前先对应了解自己需要的版本

1. Anacoda 安装

1.1. Anaconda的下载

Anaconda百度网盘的链接
提取码:jy0w

1.2. Anaconda的安装

选择Just Me —> 选择安装路径(可以不安装在C盘)—>选择Add Anaconda to my PATH environment variable

2. CUDA和cudnn的下载和安装

CUDA的百度网盘链接
提取码:w1vy

Cudnn的网盘链接
提取码:wie5

其中CUDA和Cudnn是需要相同版本的才可以

2.1. CUDA安装

2.1.1. 点击.exe文件,选择下载路径(默认路径)

img

img

img

img

2.1.2. 推荐自定义下载

img

2.1.3. 选择没有visual studio的选项

img

2.1.4. 中间安装过程略过,最后一步完成安装

2.1.5. 配置环境变量

点击设置—>搜索高级系统设置—>查看环境变量
深度学习环境配置【anaconda+CUDA+cudnn+pytorch-gpu】_第1张图片

在环境变量添加:
在这里插入图片描述

2.2. cudnn下载

2.2.1. 解压cudnn文件夹,并打开CUDA的安装目录,将cudnn三个文件夹中的内容对应加在CUDA安装目录下的文件夹中

深度学习环境配置【anaconda+CUDA+cudnn+pytorch-gpu】_第2张图片

2.3. 查看CUDA+cudnn是否配置成功,打开Anaconda Prompt,输入nvcc -V,查看信息:

深度学习环境配置【anaconda+CUDA+cudnn+pytorch-gpu】_第3张图片

3. pytorch-gpu环境配置

3.1. pytorch-gpu环境的创建

打开Anaconda Prompt 输入以下命令:

conda create –n pytorch-gpu python=3.7

3.2. pytorch-gpu环境的激活

输入以下命令:

activate pytorch-gpu

3.3. 下载对应的pytorch-gpu版本

可以根据以下的图表
深度学习环境配置【anaconda+CUDA+cudnn+pytorch-gpu】_第4张图片

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

如图所示:
深度学习环境配置【anaconda+CUDA+cudnn+pytorch-gpu】_第5张图片

3.4. 验证是否安装成功

进入python:输入以下命令

import torch
torch.cuda.is_available()

显然没有安装成功:深度学习环境配置【anaconda+CUDA+cudnn+pytorch-gpu】_第6张图片
官网给的:

conda install pytorch==1.2.0 torchvision==0.4.0 cuda100 -c pytorch

没加新源下载:去掉-c pytorch

conda install pytorch==1.2.0 torchvision==0.4.0 cuda100

深度学习环境配置【anaconda+CUDA+cudnn+pytorch-gpu】_第7张图片
好的还是失败了- -。

查了好多种方案,最后根据参考链接中添加清华源的链接:终于能把pytorch安装下来了
深度学习环境配置【anaconda+CUDA+cudnn+pytorch-gpu】_第8张图片
继续输入命令:

conda install pytorch torchvision cudatoolkit=10.0

我以为即将大功告成了,最后给我返回一个false,我人麻了
还得继续解决问题:
深度学习环境配置【anaconda+CUDA+cudnn+pytorch-gpu】_第9张图片
根据参考链接3:
下载torch地址
下载对应版本:
深度学习环境配置【anaconda+CUDA+cudnn+pytorch-gpu】_第10张图片
深度学习环境配置【anaconda+CUDA+cudnn+pytorch-gpu】_第11张图片
检验成功,此刻激动的心颤抖的手,花费了半天:

深度学习环境配置【anaconda+CUDA+cudnn+pytorch-gpu】_第12张图片

原来深度学习刚入门配个环境就那么难,这毕设是才刚刚开始调研呐!!~~~~~~~~~~~~~


参考链接:
深度环境配置PyTorch=1.7.1
添加清华源
torch.cuda.is_available()返回false——解决办法
PyTorch中torch、torchvision、torchaudio版本对应关系

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