软硬件配置介绍
- 操作系统:Windows 10 和 Ubuntu 20.04 均适用
- GPU:Nvidia Geforce RTX 3060
- Python:3.8
- Tensorflow-gpu: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),使用以下内容替换配置文件并保存。
[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,并激活该环境。文章来源:https://www.toymoban.com/news/detail-409730.html
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'))
若最后输出以下内容,则配置成功。
否则请检查Python、TensorFlow、cudatoolkit和cudnn各自的版本是否与本文一致。文章来源地址https://www.toymoban.com/news/detail-409730.html
到了这里,关于深度学习环境配置的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!