在配置之前先对应了解自己需要的版本
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文件,选择下载路径(默认路径)
2.1.2. 推荐自定义下载
2.1.3. 选择没有visual studio的选项
2.1.4. 中间安装过程略过,最后一步完成安装
2.1.5. 配置环境变量
点击设置—>搜索高级系统设置—>查看环境变量
在环境变量添加:
2.2. cudnn下载
2.2.1. 解压cudnn文件夹,并打开CUDA的安装目录,将cudnn三个文件夹中的内容对应加在CUDA安装目录下的文件夹中
2.3. 查看CUDA+cudnn是否配置成功,打开Anaconda Prompt,输入nvcc -V,查看信息:
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 install pytorch torchvision cudatoolkit=10.0 -c pytorch
如图所示:
3.4. 验证是否安装成功
进入python:输入以下命令
import torch
torch.cuda.is_available()
显然没有安装成功:
官网给的:
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
好的还是失败了- -。查了好多种方案,最后根据参考链接中添加清华源的链接:终于能把pytorch安装下来了
继续输入命令:
conda install pytorch torchvision cudatoolkit=10.0
我以为即将大功告成了,最后给我返回一个false,我人麻了
还得继续解决问题:
根据参考链接3:
下载torch地址 下载对应版本:
检验成功,此刻激动的心颤抖的手,花费了半天:
原来深度学习刚入门配个环境就那么难,这毕设是才刚刚开始调研呐!!~~~~~~~~~~~~~
参考链接:
深度环境配置PyTorch=1.7.1添加清华源torch.cuda.is_available()返回false——解决办法PyTorch中torch、torchvision、torchaudio版本对应关系