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

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文件,选择下载路径(默认路径)

conda 安装tensorflow GPU_深度学习

conda 安装tensorflow GPU_环境变量_02

conda 安装tensorflow GPU_CUDA_03

conda 安装tensorflow GPU_CUDA_04

2.1.2. 推荐自定义下载

conda 安装tensorflow GPU_CUDA_05

2.1.3. 选择没有visual studio的选项

conda 安装tensorflow GPU_CUDA_06

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

2.1.5. 配置环境变量

点击设置—>搜索高级系统设置—>查看环境变量

conda 安装tensorflow GPU_python_07

在环境变量添加:

conda 安装tensorflow GPU_python_08

2.2. cudnn下载

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

conda 安装tensorflow GPU_pytorch_09

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

conda 安装tensorflow GPU_CUDA_10

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版本

可以根据以下的图表

conda 安装tensorflow GPU_pytorch_11

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

如图所示:

conda 安装tensorflow GPU_环境变量_12

3.4. 验证是否安装成功

进入python:输入以下命令

import torch
torch.cuda.is_available()

显然没有安装成功:

conda 安装tensorflow GPU_CUDA_13


官网给的:

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

conda 安装tensorflow GPU_python_14


好的还是失败了- -。查了好多种方案,最后根据参考链接中添加清华源的链接:终于能把pytorch安装下来了

conda 安装tensorflow GPU_环境变量_15


继续输入命令:

conda install pytorch torchvision cudatoolkit=10.0

我以为即将大功告成了,最后给我返回一个false,我人麻了

还得继续解决问题:

conda 安装tensorflow GPU_python_16


根据参考链接3:

下载torch地址 下载对应版本:

conda 安装tensorflow GPU_CUDA_17


conda 安装tensorflow GPU_深度学习_18


检验成功,此刻激动的心颤抖的手,花费了半天:

conda 安装tensorflow GPU_深度学习_19

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


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