当前位置:   article > 正文

ubunut22.04安装深度学习框架(记录)_deepspeed ubuntu

deepspeed ubuntu

1.安装anaconda

网站Anaconda | Anaconda Distribution 安装最新版本conda

然后进入下载的文件目录中,打开终端,执行sh文件(遇到更多就不断回车),安装到默认目录

  1. # 改以下名字
  2. bash Anaconda3-2021.11-Linux-x86_64.sh

2.安装pycharm

PyCharm官网

官网上下载一个社区版本的文件,进入文件所在目录,提取出pycharm-community-2022.2.3;进入其中的bin目录,执行pycharm.sh

./pycharm.sh

安装完成之后,会弹出来界面,类似下图。此处可以设置添加桌面图标(默认没有桌面图标,只能每次sh)

3.安装cuda和cudnn

1.首先查看电脑支持的cuda版本

nvidia-smi

会出现如下结果,显示最高支持cuda为11.7

先查看一下ubuntu22.04是否直接支持cuda11.7(不需要手动安装其他依赖),CUDA Toolkit Archive | NVIDIA Developer

然后看一下pytorch和mmcv是否有对应版本https://download.openmmlab.com/mmcv/dist/cu113/torch1.12.0/index.html

 2.安装cuda

官网直接搜cuda11.7,按照官网代码安装

安装完成后,环境编译,但是cuda11.7不确定是否有必要(因为.bashrc里有一些类似的内容)

  1. sudo gedit ~/.bashrc
  2. #最后添上:
  3. export CUDA_HOME=/usr/local/cuda-11.6
  4. export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
  5. export PATH=${CUDA_HOME}/bin:${PATH}
  6. source ~/.bashrc #路径生效

3.安装cudnn

CUDA Deep Neural Network (cuDNN) | NVIDIA Developer

 由于没有对应的CUDA11.7对应的cuDNN版本,暂时使用cuDNN对应CUDA11.x版本的。

下载并解压,进入解压后的文件夹打开终端 

  1. sudo cp include/cudnn* /usr/local/cuda/include
  2. sudo cp lib/libcudnn* /usr/local/cuda/lib64
  3. sudo chmod a+r /usr/local/cuda/include/cudnn*
  4. sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

cuDNN安装完成,通过如下命令查看cudnn下版本信息

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

注意:如果遇到如下问题

问题一:

 参考W: GPG error: cudnn GPG key 错误_迷途-归处的博客-CSDN博客

简单过程就是

  1. # 改序号
  2. sudo cp /var/cudnn-local-repo-ubuntu2204-8.5.0.96/*.gpg /usr/share/keyrings/

问题二:

 首先进入仓库列表

cd /etc/apt/sources.list.d

用眼看和报错仓库名字一样的文件,执行

  1. #*替换成对应名字
  2. sudo rm *.list

最后

  1. sudo apt-get update
  2. sudo apt-get upgrade

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/黑客灵魂/article/detail/849254
推荐阅读
相关标签
  

闽ICP备14008679号