确定显卡和驱动版本

CUDA 12.2 Release Notes

考虑到后续要安装pytorch,选择pytorch支持cuda的最高版本cuda11.8

一、cuda的安装

1. 首先下载所匹配的CUDA Toolkit(这里选择CUDA Toolkit 11.8.0

CUDA Toolkit Archive | NVIDIA Developer

 选择对于的系统版本,复制第一条命令到命令行;复制第二条命令到cuda安装包所在目录下执行

wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.runsudo 
sh cuda_11.8.0_520.61.05_linux.run

注意:

安装时出现多个驱动的问题:Existing package manager installation of the driver found. It is strongly recommended that you remove this before continuing.

解决:sudo sh cuda_11.8.0_520.61.05_linux.run --toolkit --silent --override

2. 添加环境变量

gedit ~/.bashrc 在末尾添加下面两行:

export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-11.8/bin/:$PATH

保存关闭命令和输入source ~/.bashrc

 3. 验证cuda是否安装成功

nvcc --version

二、cudnn的安装

https://developer.nvidia.com/rdp/cudnn-archive

先注册账号,然后选择Download cuDNN v8.9.2 (June 1st, 2023), for CUDA 12.x

根据官方教程安装并验证https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html 

1.1.3. Installing Zlib

 1.3.2. Debian Local Installation

1.4. Verifying the Install on Linux

注意:

1. 安装过程中8.x.x.x-1.cudaX.Y替换为对应的cuda版本即可,这里是8.9.3.28-1+cuda11.8

2. make clean && make 出现错误

test.c:1:10: fatal error: FreeImage.h: No such file or directory
    1 | #include "FreeImage.h"
      |          ^~~~~~~~~~~~~
compilation terminated.
解决:sudo apt-get install libfreeimage3 libfreeimage-dev

重新编译后,运行即可。

三、Anaconda安装

1. https://www.anaconda.com/

https://www.anaconda.com/点击download,下载完成后在下载目录打开终端输入:

sudo apt-get install libgl1-mesa-glx libegl1-mesa libxrandr2 libxrandr2 libxss1 libxcursor1 libxcomposite1 libasound2 libxi6 libxtst6

sh 安装包名

source ~/.bashrc

conda update --all

安装完成!!

2. 创建桌面快捷方式

桌面打开终端窗口,输入gedit anaconda.desktop

将下面内容复制并改写

[Desktop Entry]
Name=Anaconda
Version=1.0
Type=Application
Exec=/home/lc/anaconda3/bin/anaconda-navigator
Icon=/home/lc/anaconda3/pkgs/anaconda-navigator-2.4.2-py311h06a4308_0/lib/python3.11/site-packages/anaconda_navigator/static/images/anaconda-icon-256x256.png
Terminal=false
StartupNotify=true

可以看到桌面多了个快捷方式,右键->允许运行,双击即可打开。

四、Pycharm安装

下载安装包https://www.jetbrains.com/pycharm/download/?section=linux

剪切到主目录,进入~/pycharm-community-2023.1.4/bin,打开终端输入 ./pycharm.sh

安装成功!

创建快捷方式:打开pycharm,设置->点击create desktop entry即可

汉化:

打开pychar,plugins,搜索chinese(Simpified)安装即可

五、pytorch安装

终端输入python查看python版本(这里是3.11)

终端输入conda create -n pytorch python=3.11在anaconda里创建pytorch环境

安装pytorch:https://pytorch.org/

 选择好环境和版本,生成命令,在终端中执行:

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

安装完毕!

检查pytorch是否成功安装:

 说明安装成功!!

六、Jupyter安装

anaconda安装过程中会自带安装jupyter,只不过是在base环境中的,如果想在pytorch环境中安装jupyter的话,首先查看base环境中的包conda list,pytorch环境中缺少ipykernel,conda install nb_conda进行安装,之后jupyter notebook启动jupyter,新建选择所创建的pytorch环境。

检查是否可用,shift+回车运行当前命令行

Logo

瓜分20万奖金 获得内推名额 丰厚实物奖励 易参与易上手

更多推荐