Ubuntu20.04安装Nvidia显卡驱动教程以及深度学习环境搭建

前言

记录一下新系统环境安装的过程,如果前面没有删除干净的,可以查看最下面参考文章中的链接。

Nvidia显卡驱动教程

先查询电脑最适合的显卡驱动版本

python 复制代码
ubuntu-drivers devices

最佳显卡驱动版本为nvidia-driver-535-server-open,但我自己下载后重启发现没有找到设备,安装下面的问题就解决了。

python 复制代码
sudo apt install nvidia-driver-535

如果你不需要服务器优化版本,可以选择nvidia-driver-535。这是标准版本的驱动,适合大多数桌面和工作站环境。

安装后重启

python 复制代码
sudo reboot

然后输入验证即可

python 复制代码
nvidia-smi

我这里的设备装有两块NVIDIA GeForce RTX 3090显卡,CUDA版本为12.2。

安装miniconda

进入这里清华大学开源软件镜像站 | Tsinghua Open Source Mirror,搜索anaconda,进入后选择miniconda。

找到适合你的Linux系统版本的Miniconda安装程序,右键复制链接地址。

打开终端,利用wget命令下载安装程序文件:

python 复制代码
wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniconda3-py38_4.9.2-Linux-x86_64.sh

接着,运行即可

python 复制代码
bash Miniconda3-py38_4.9.2-Linux-x86_64.sh

在安装脚本时候遇到权限问题,可尝试使用下面的指令

python 复制代码
chmod +x Miniconda3-py38_4.9.2-Linux-x86_64.sh

如果出现conda:未找到命令

python 复制代码
source ~/.bashrc

创建虚拟环境

查看当前的虚拟环境

python 复制代码
conda env list

创建新环境

python 复制代码
conda create -n zjrpy38 python=3.8

激活环境

python 复制代码
conda activate zjrpy38

退出环境:deactivate,我通常会省去这步,直接activate 其他的虚拟环境名,这样方便转换到其他虚拟环境。

删除虚拟环境:

复制代码
conda remove -n zjrpy38 --all

下载第三方库

torch请从此处查看Previous PyTorch Versions | PyTorch

请注意,这里的cuda版本应该比你当前的设备要小,最好不要用相同版本的。

python 复制代码
pip install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 --index-url https://download.pytorch.org/whl/cu121

一些可能会用到的

pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple

pip install scikit-image -i https://pypi.tuna.tsinghua.edu.cn/simple

pip install matplotlib -i https://pypi.tuna.tsinghua.edu.cn/simple

安装Pycharm

pycharm的安装包下载网址:https://www.jetbrains.com/zh-cn/pycharm/download/#section=linux

解压之后

你可以进入下面的bin文件夹

python 复制代码
cd /home/ecgserver2010/AI/pycharm/pycharm-community-2024.3.2/bin

然后输入

python 复制代码
sh pycharm.sh

每次这样不太方便,这里我们来建立桌面快捷方式,先进入下面的目录,这是存储应用程序快捷方式的目录。

python 复制代码
cd /usr/share/applications/

使用文本编辑器创建一个新的 .desktop 文件

复制代码
sudo nano pycharm.desktop

在编辑器中添加以下内容:

python 复制代码
[Desktop Entry]
Name=PyCharm
Comment=Python IDE
Exec=/home/ecgserver2010/AI/pycharm/pycharm-community-2024.3.2/bin/pycharm.sh
Icon=/home/ecgserver2010/AI/pycharm/pycharm-community-2024.3.2/bin/pycharm.png
Terminal=false
Type=Application
Categories=Development;IDE;

这里应当使用你自己的PyCharm路径和图标路径

保存并关闭文件,使用 Ctrl + X 保存。

接下来就可以在应用搜索框中找到了:

参考文章

Ubuntu20.04安装Nvidia显卡驱动教程-CSDN博客

深度学习环境搭建入门环境搭建(pytorch版本)_pytorch环境搭建-CSDN博客

在Ubuntu上安装配置pycharm(linux)_ubuntu pycharm-CSDN博客

相关推荐
测试界的海飞丝11 小时前
10道软件测试面试题及其答案:
服务器·测试工具·职场和发展
t1987512811 小时前
在Ubuntu 22.04系统上安装libimobiledevice
linux·运维·ubuntu
skywalk816311 小时前
linux安装Code Server 以便Comate IDE和CodeBuddy等都可以远程连上来
linux·运维·服务器·vscode·comate
@游子11 小时前
内网渗透笔记-Day5
运维·服务器
晚风吹人醒.12 小时前
缓存中间件Redis安装及功能演示、企业案例
linux·数据库·redis·ubuntu·缓存·中间件
Yawesh_best12 小时前
告别系统壁垒!WSL+cpolar 让跨平台开发效率翻倍
运维·服务器·数据库·笔记·web安全
V***u45312 小时前
MS SQL Server partition by 函数实战二 编排考场人员
java·服务器·开发语言
Hard but lovely12 小时前
linux: pthread库的使用和理解
linux
麦麦鸡腿堡13 小时前
Java_网络编程_InetAddress类与Socket类
java·服务器·网络
牛哥带你学代码13 小时前
服务器运行常用指令
运维·服务器