Windows10、Ubuntu20.04深度学习环境配置(CUDA安装、CUDNN安装、3060显卡)

一、软硬件配置介绍

  • 操作系统:Windows 10 和 Ubuntu 20.04 均适用
  • GPU:Nvidia Geforce RTX 3060
  • Python:3.8
  • Tensorflow:2.5.0

二、环境配置步骤

1、安装显卡驱动

(1)Windows 10

​ 在Nvidia驱动下载官网下载522.25版本的驱动并安装。

(2)Ubuntu 20.04

​ 打开系统设置,找到软件更新里的附加驱动,选择470版本的驱动即可。注意,安装完成后需要重启一次。

​ 安装完成后使用 nvidia-smi 命令进行验证。

2、安装并配置Conda

(1)安装conda

​ 此处选择miniconda。访问miniconda官网下载对应版本并安装。注意:需要加入path环境变量。

(2)conda配置国内源

配置conda清华源:

​ 首先执行以下命令生成conda配置文件:

conda config --set show_channel_urls yes

​ 在用户目录下找到配置文件.condarc,打开用以下内容替换并保存。

channels:
  - defaults
show_channel_urls: true
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch-lts: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
配置pip清华源:

​ 在用户目录下新建目录pip(Ubuntu下为.pip),在该目录下新建配置文件pip.ini(Ubuntu下为pip.conf),使用以下内容替换配置文件并保存。

​ Windows使用以下内容:

[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
[install]
trusted-host = pypi.tuna.tsinghua.edu.cn

​ Ubuntu使用以下内容:

[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
[install]
trusted-host = https://pypi.tuna.tsinghua.edu.cn

3、安装cuda和cudnn

​ 首先创建名为test的conda环境(名字可以自己改),python版本为3.8,并激活该环境。

conda create -n test python=3.8
conda activate test

(1)安装cuda和cudnn

conda install cudatoolkit==11.3.1
conda install cudnn==8.2.1

(2)验证

import tensorflow as tf

print(tf.test.is_gpu_available())
print(tf.config.list_pysical_list('GPU'))

热门相关:地球第一剑   霸皇纪   大神你人设崩了   寂静王冠   大神你人设崩了