快速搭建ubuntu下深度学习环境导航（从装机到pytorch+cuda）

byx0288

已于 2023-04-22 17:12:44 修改

阅读量2w

点赞数

文章标签： ubuntu 深度学习 linux Powered by 金山文档

于 2023-03-17 22:11:03 首次发布

本文链接：https://blog.csdn.net/byx0288/article/details/129628738

版权

移动硬盘安装ubuntu

(167条消息) 移动固态+uefi引导+ubuntu20.04安装方法_byx0288的博客-CSDN博客

环境构建步骤

apt换源

(167条消息) Ubuntu更换清华源 apt_apt 清华源_宇脩的博客-CSDN博客

vim安装

sudo apt-get install vim

gcc库安装

使用命令

sudo apt install build-essential
gcc -V

该命令将安装一堆新包，包括gcc，g ++和make。

检查是否安装成功

cuda以及cudnn安装

首先需要关闭bios的安全启动模式，即Secure Boot模式

(167条消息) ERROR: The kernel module failed to load, because it was not signed by a key that is trusted by the_error: the kernel module failed to load. secure bo_yiyayi@che的博客-CSDN博客

由于ubuntu自带nouveau驱动会与nvidia驱动产生冲突导致安装失败所以我们需要先禁用该驱动

sudo vim /etc/modprobe.d/blacklist-nouveau.conf

在文本最后加入

blacklist nouveau
options nouveau modeset=0

更新使其生效

sudo update-initramfs -u

重启电脑

reboot

(167条消息) Ubuntu 安装 NVIDIA 显卡驱动详细步骤（ERROR: The Nouveau kernel driver is currently in use by your system）_ubuntu安装nvidia驱动_wohu1104的博客-CSDN博客

安装必要的依赖库

否则在安装cudnn环节会出现错误

sudo apt-get install libfreeimage3 libfreeimage-dev

https://www.yellowei.com/wei-huang/426/

开始安装cuda以及cudnn

cuda安装参照如下，可以直接依照官方使用wget命令进行安装包的下载

(167条消息) Ubuntu下CUDA的安装及配置(run方式)_ubuntu安装cuda_花花少年的博客-CSDN博客

cudnn安装使用debian的会更方便些，参照如下官方文档即可

Installation Guide :: NVIDIA Deep Learning cuDNN Documentation

miniconda安装

依照如下安装最新miniconda

(167条消息) 在Ubuntu上安装和使用miniconda_ubuntu安装miniconda_黄瓜炒肉的博客-CSDN博客

torch检测

import torch
print(torch.cuda.is_available())
# 返回True 接着用下列代码进一步测试
print(torch.zeros(1).cuda())