深度学习环境安装

前置知识

1.国内显卡厂商如技嘉、七彩虹等与英伟达的关系

英伟达开发的是显卡核心**(GPU),然后把这些核心出售给技嘉、七彩虹等厂商,这些厂商再为显卡核心设计PCB板,供电,散热等,最终制造出一块完整的、即插即用的显卡**

2.GPU常见的厂家

NVIDIA英伟达、AMD

Windows

安装虚拟环境

1.Anaconda官网地址:https://www.anaconda.com/download#downloads 这个应该比较好装,网上的教程应该不会出错

2.在DOS界面(命令行黑框界面)创建虚拟环境 conda create - -name 虚拟环境名字 python=3.11

3.激活刚才创建的虚拟环境 conda activate 虚拟环境名字
在这里插入图片描述

下载cuda

1.进入NVIDIA官网下载驱动:https://www.nvidia.cn/geforce/drivers/

在这里插入图片描述

2.更新完显卡驱动以后进入命令行界面,输入nvidia-smi,可以看见自己显卡的相关信息。

在这里插入图片描述

cuda安装的版本不要超过这个 CUDA Version:12.3 。显示的是驱动支持的最大cuda版本。并不是当前安装的

3.进入cuda官网选择下载适配自己电脑显卡的CUDA Toolkit工具包:https://developer.nvidia.com/cuda-toolkit-archive

在这里插入图片描述

4.安装cuda

在这里插入图片描述

推荐默认安装路径,除了以下情况都是点击下一步

在这里插入图片描述

在这里插入图片描述

在CUDA下,取消勾选Visual Studio,因为就算选择安装VS,很大概率也会安装失败。然后点击下一步。

在这里插入图片描述

5.完成安装,检查环境变量中是否添加了路径,如果没有,则需要自己手动添加。

在这里插入图片描述

手动添加环境变量

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\libnvvp

6.测试环境是否安装成功

命令行输入nvcc --version,若显示已安装的cuda版本,则表示安装完成。

在这里插入图片描述

下载cuDNN

1.进入cuDNN官网:https://developer.nvidia.com/rdp/cudnn-archive

2.根据之前下载的cuda版本选择对应的cuDNN

在这里插入图片描述

3.根据电脑选择对应的压缩包

在这里插入图片描述

4.解压cuDNN,解压后有三个文件夹,把三个文件夹拷贝到cuda的安装目录(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8)下,替换已存在的文件即可。

在这里插入图片描述

5.添加环境变量

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\include

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\lib

下载pytorch

1.pytorch官网:https://pytorch.org/

2.选择对应的版本,将Run this Command中的代码复制,黏贴到命令行中运行。等待安装即可。

在这里插入图片描述

检测是否安装成功

1.命令行中输入python

外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传

2.导入torch包,检查gpu是否可用。如果返回True,就代表安装成功了。

import torch
torch.cuda.is_available()

在这里插入图片描述

3.需要卸载pytorch的话

pytorch使用什么方式安装的,就使用什么方式卸载。

①执行conda方式

conda uninstall pytorch
conda uninstall libtorch

②执行pip方式

pip uninstall torch

Linux

安装虚拟环境

下载anaconda

使用以下命令进行下载

wget https://repo.anaconda.com/archive/Anaconda3-2023.03-Linux-x86_64.sh
如果服务器网不行的话可以在本机上先下载下来然后再传到服务器上anaconda官网https://www.anaconda.com/download/success
在这里插入图片描述

安装anaconda

bash Anaconda3-2023.03-Linux-x86_64.sh
一直回车确认,出现这个说明安装成功
在这里插入图片描述

配置环境变量

在/etc/profile文件的末尾添加如下语句

export PATH=$PATH:/root/anaconda/bin(anaconda的安装路径)

重新载入配置文件

source /etc/profile

创建虚拟环境

conda create - -name 虚拟环境名字 python=3.11

激活虚拟环境

conda activate 虚拟环境名字

下载cuda

使用命令nvidia-smi查看显卡

查看虚拟环境命令

conda env list

在这里插入图片描述

退出当前的虚拟环境

conda deactivate

删除虚拟环境

conda remove -n 需要删除的环境名 --all

  • 13
    点赞
  • 19
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值