深度学习环境配置

发布时间：2023年12月19日

一、Anaconda安装

下载：从清华大学开源软件镜像下载
镜像网址
在这里插入图片描述

出现base即为安装成功：

在这里插入图片描述

检查显卡的驱动是否正确安装：

（GPU可以显示出名称）
在这里插入图片描述

·GPU0是集显=集成显卡是主板自带的显卡。
·GPU1是独显即独立显卡，是单独的一张显卡，性能一般会比集显要高。

管理环境

用conda指令创建一个pytorch环境

conda create -n pytorch python=3.8

激活

conda activate pytorch

在这里插入图片描述

出现的错误：
在这里插入图片描述
解决方法：
（1）检查环境变量
在这里插入图片描述
path
在这里插入图片描述
（2）恢复默认源

conda config --remove-key channels

检查环境中的工具包：

pip list

在这里插入图片描述

二、下载cuda

cuda下载链接
在这里插入图片描述
在这里插入图片描述

环境配置：

在这里插入图片描述
检查CUDA是否安装成功：

nvcc -V

在这里插入图片描述
进入C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\extras\demo_suite（或者是自己安装的CUDA路径），将bandwidthTest.exe和deviceQuery.exe拖入到cmd中执行，若结果为都为PASS，表示CUDA的安装与环境变量配置均成功。
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

三、下载cuDNN

cuDNN下载链接
（1）注册NVIDIA开发者账号
（2）下载
在这里插入图片描述
（3）解压下载好的安装包，cudnn-windows-x86_64-8.9.7.29_cuda11-archive.zip，包含三个文件夹。
在这里插入图片描述
（4）将三个文件全部复制到cuda安装目录下的同名文件夹中，即可完成安装。
在这里插入图片描述

四、下载pytorch

进入虚拟环境pytorch

激活

conda activate pytorch

安装pytorch

pytorch官网
在这里插入图片描述
复制安装命令

将指令输入到虚拟环境中，进行安装。

在这里插入图片描述
出现的错误：
在这里插入图片描述
因为下载时间过长，会自动断开，就会提示连接失败，设置延时1000s。

conda config --set remote_read_timeout_secs 1000.0

再安装pytorch
在这里插入图片描述

检查pytorch是否安装成功

输入python进入python环境

import torch

torch.cuda.is_available()

显示True，说明配置成功。
在这里插入图片描述
出现错误：
在这里插入图片描述
要注意显卡、cuda和pytorch的版本一致。
用清华源安装的版本可能是cpu版本的，要注意！！

检查显卡GPU和驱动版本的方法：

nvidia-smi

在这里插入图片描述

文章来源:https://blog.csdn.net/shi_jiaye/article/details/134999326
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
最新文章