个人记录-2022.4.27-给3090,ubuntu20.04配置深度学习环境

个人记录-2022.4.27-给3090,ubuntu20.04配置深度学习环境,第1张

参考教程(实际 *** 作时略有不同):
1

2

3

在老师电脑上的版本说明:
在conda下新建了名为tf-gpu的虚拟环境
python 3.9
tensorflow-gpu 2.5.0
cudatoolkit 11.3.1
cudnn 8.2.1
keras 2.5.0

实际 *** 作命令行记录:
#查看显卡信息,发现系统未检测到显卡
nvidia-smi
lshw -c video

#查看对应的显卡驱动版本并安装
sudo apt-get --purge remove nvidia*
sudo apt autoremove
ubuntu-drivers devices
sudo apt install nvidia-driver-510-server

#重启电脑

#创建conda虚拟环境tf-gpu
python
conda create -n tf-gpu python==3.9
conda info -e

#激活虚拟环境tf-gpu
source activate tf-gpu
conda list

#输入以下命令打开condarc文件
vi ~/.condarc

#然后输入以下内容
该方法参考自清华源anaconda

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

#然后在tf-gpu该虚拟环境中安装库
source activate tf-gpu
conda install cudatoolkit=11.3 cudnn=8.2.1

#这里选择清华、阿里镜像源,或不选择源直接安装,网速较快(以下三种命令选其一即可)
pip install --default-time=300 tensoflow-gpu2.5.0 keras2.5.0rc0 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install --default-time=300 tensorflow-gpu2.5.0 keras2.5.0rc0 -i https://mirrors.aliyun.com/pypi/simple
pip install --default-time=300 tensorflow-gpu2.5.0 keras2.5.0rc0

#在终端输入以下命令行验证
source activate tf-gpu
python
import tensorflow as tf
print(tf.version,tf.test.is_gpu_available())

#显示结果为2.5.0 True,安装成功

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/786588.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-05-05
下一篇 2022-05-05

发表评论

登录后才能评论

评论列表(0条)

保存