Ubuntu深度环境搭建

先贴几个参考的帖子:

Ubuntu20.04下深度学习环境配置(持续维护)_ubuntu深度学习-CSDN博客

前置内容如换源、GPU驱动、gcc、anaconda等都已配置好,只写怎么在已有环境下搭建新的云·深度学习代码运行环境

1、创建新环境:conda create -n litimono python=3.8

【ps:conda activate litimono,激活环境,若失败,则使用source activate litimono激活环境】

2、安装pytorch:conda install pytorch==1.11.0 torchvision torchaudio cudatoolkit=10.2 -c pytorch

【此处根据自己的需要做更改】

        验证:

        $python \\进入python执行界面

        $import torch \\若无任何提示,安装成功

        $print(torch.__version__) \\查看pytorch版本,出现1.11.0则表示安装成功

        $exit() \\退出python执行界面

3、安装cuda,若之前已经配置过环境,cuda应该是安装好了的,在环境下使用nvcc --version检查,若出现下图则表明cuda安装成功,否则参考上述链接安装cuda

4、安装cudnn同上,若之前已经配置过环境,cudnn应该是安装好了的,在进行完 “2、pytorch安装” 后用torch.cuda.is_available()进行验证的,若出现Ture则表示安装成功,否则参考上述链接安装cudnn。

5.litimono复现过程

litemono代码_linear_warmup_cosine_annealing_warm_restarts_weigh-CSDN博客

在上面的步骤完成后,进行以上文章的过程5--在终端使用

python -c "import site; print(site.getsitepackages())

来确定文件位置,之后将下载解压的文件夹按文章说明放到该位置即可。

  • 10
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Ubuntu上搭建深度学习环境,需要以下步骤: 1. 安装CUDA和cuDNN CUDA是NVIDIA提供的用于GPU加速的并行计算平台,cuDNN是NVIDIA提供的深度学习库。在安装之前,需要确认你的GPU型号是否支持CUDA和cuDNN,具体可以查看官网的支持列表。安装CUDA和cuDNN的步骤如下: - 安装CUDA:可以从官网下载对应的CUDA安装包,然后按照提示进行安装。 - 安装cuDNN:需要先注册NVIDIA开发者账号,然后从官网下载对应的cuDNN安装包,然后按照提示进行安装。 2. 安装Python和相关库 深度学习常用的编程语言是Python,需要安装Python和相关的库。一般来说,可以通过anaconda来安装Python和相关库,具体步骤如下: - 下载anaconda安装包:可以从官网下载对应的anaconda安装包。 - 安装anaconda:打开终端,进入下载目录,执行以下命令: ``` bash Anaconda3-2020.11-Linux-x86_64.sh ``` - 创建Python虚拟环境:执行以下命令: ``` conda create -n env_name python=3.6 ``` 其中env_name为虚拟环境的名称。 - 激活虚拟环境:执行以下命令: ``` conda activate env_name ``` - 安装深度学习相关库:执行以下命令: ``` conda install tensorflow-gpu keras pandas matplotlib jupyter ``` 其中tensorflow-gpu是支持GPU加速的TensorFlow库,keras是一个高级神经网络API,pandas是Python中数据处理库,matplotlib是Python中的绘图库,jupyter是一个交互式的笔记本。 3. 安装其他工具 除了上述步骤,还需要安装其他常用的工具,比如git、vim等。可以通过apt-get命令来安装,执行以下命令: ``` sudo apt-get update sudo apt-get install git vim ``` 安装完成后,就可以开始使用深度学习环境进行开发了。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值