深度学习环境安装经验总结_深度学习软件安装

一、安装miniconda

1.1 安装地址

https://mirror.tuna.tsinghua.edu.cn/anaconda/miniconda/
1

1.2 安装版本

x86_64:对应电脑版本是64位的
x86:对应电脑版本是32位的
1
2

1.3 安装步骤

一直点击下一步即可。

注意一下步骤勾选

1.4 测试是否成功

// 在命令行界面输入
conda
1
2

二、查看显卡设备

2.1 查看显卡

在搜索框中输入控制面板

2.2 查看显卡对应的CUDA版本

一般情况下电脑重装以后，框选区域为最小的版本。并且在安装CUDA驱动后会随之更新。

三、安装英伟达驱动

https://www.nvidia.cn/Download/index.aspx?lang=cn
1

下载完毕之后就又是很简单的下一步下一步直到完成，完成之后，在cmd中输入命令

nvidia-smi
1

四、安装CUDA、cuDNN

CUDA与cuDNN的版本要一致

// CUDA地址
https://developer.nvidia.com/cuda-toolkit-archive
//cuDNN地址
https://developer.nvidia.com/cudnn
1
2
3
4

4.1 下载和安装CUDA

4.1.1下载

4.1.2 安装

前面已经安装过驱动了，因此这里不用勾选。

然后一直下一步即可

nvcc -V
1

4.2 下载和安装cuDNN

下载cuDNN需要注册英伟达账号+科学上网

4.2.1 下载

4.2.2 安装

下载好后解压缩

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
1

然后把解压缩以后的文件放置在上述文件夹下

在C:\Program Files\NVIDIA GPU Computing \Toolkit\CUDA\v10.1\extras\demo_suite这个目录下打开cmd，输入deviceQuery.exe，如果结果为pass证明安装成功。
在这里插入图片描述

五、安装pytorch和测试gpu是否可用

5.1 注意pytorch和CUDA的版本要一致或兼容

5.2 安装pytorch的准备工作

应该下载与CUDA10.1兼容的pytorch

pytorch安装地址

https://pytorch.org/get-started/previous-versions/
1

安装指令

conda create -n pytorch_gpu python=3.7
conda activate pytorch_gpu
pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.6.0 torchvision==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
# pip install torch==1.6.0 torchvision==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
1
2
3
4

5.3 测试

import torch
print(torch.__version__)
print(torch.cuda.is_available())#cuda是否可用
torch.cuda.device_count()#返回GPU的数量
torch.cuda.get_device_name(0)#返回gpu名字，设备索引默认从0开始
1
2
3
4
5