深度学习环境搭建

yyybeautiful

已于 2024-04-10 20:35:09 修改

阅读量570

点赞数 1

文章标签：深度学习 python 人工智能

于 2022-10-11 20:23:14 首次发布

本文链接：https://blog.csdn.net/yyybeautiful/article/details/127270623

版权

1. 安装Anaconda

清华大学开源镜像：https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
安装2022.5月

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-Y7se4R18-1665490107615)(C:\Users\Administrator\AppData\Roaming\Typora\typora-user-images\image-20221006155246807.png)]$

输入conda --version

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-SjJr58cJ-1665490107617)(C:\Users\Administrator\AppData\Roaming\Typora\typora-user-images\image-20221006190229769.png)]$
则安装成功

2. 安装 CUDA 11.7

安装地址：https://developer.nvidia.com/cuda-toolkit-archive

CUDA需和cuDANN版本对应

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-D9drNueG-1665490107619)(C:\Users\Administrator\AppData\Roaming\Typora\typora-user-images\image-20221006160204897.png)]$
在这里插入图片描述

3. 安装cuDANN

官网： https://developer.nvidia.com/rdp/cudnn-archive

在这里插入图片描述

4. 三个下载完毕

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-OOE0XNhD-1665490107622)(C:\Users\Administrator\AppData\Roaming\Typora\typora-user-images\image-20221006161814208.png)]$

5. 安装Anaconda

可选择个人

全选上

在这里插入图片描述

验证输入python，出现python的版本即可

6. 安装cuda

第一个路径默认（后面系统会自动删掉）

选择精简

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-oB3QfHVD-1665490107623)(C:\Users\Administrator\AppData\Roaming\Typora\typora-user-images\image-20221006162051844.png)]$

安装完毕在命令行输入nvcc --version，可以看到我安装的是11.7

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-hxJP9Gyg-1665490107623)(C:\Users\Administrator\AppData\Roaming\Typora\typora-user-images\image-20221006162540151.png)]$

7. 安装cuDANN

解压cuDANN的压缩包发现里面会有三个文件夹 bin,include,lib

将它们复制到cuda安装下面C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.17，这是默认的路径

在这里插入图片描述

在路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7\extras\demo_suite中，用cmd运行bandwidthTest.exe和deviceQuery.exe，如果得到两个PASS就证明成功了，如图。

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-SnuwhJrY-1665490107625)(C:\Users\Administrator\AppData\Roaming\Typora\typora-user-images\image-20221006163029933.png)]$

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-NfmKHKi5-1665490107626)(C:\Users\Administrator\AppData\Roaming\Typora\typora-user-images\image-20221006163113441.png)]$

8. 换源

conda config --remove-key channels
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
pip config set global.index-url https://mirrors.ustc.edu.cn/pypi/web/simple

9. 下载pytorch 1.12.1

去官网Start Locally | PyTorch

拷贝

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

等待下载（很慢）

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-fpj28FBW-1665490107626)(C:\Users\Administrator\AppData\Roaming\Typora\typora-user-images\image-20221006191803864.png)]$

验证(进入Anaconda命令)

pip list

里面有torch 1.12.1

python
import torch
torch.cuda.is_available()
显示true

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-iRFhwHiY-1665490107627)(C:\Users\Administrator\AppData\Roaming\Typora\typora-user-images\image-20221006194857814.png)]$