Linux系统一键配置conda、cuda、pytorch

配置conda

AnacondaMiniconda。两者的主要区别如下:

特性AnacondaMiniconda
安装包大小较大(约 3 GB)很小(约 50 MB)
预装包包含 1500+ 科学计算包(如 NumPy、Pandas、Jupyter)仅包含 Conda、Python 及少量核心依赖
适用场景适合初学者或需要立即使用完整数据科学套件的用户适合希望轻量安装、按需自定义环境的用户
灵活性开箱即用,但环境较“重”高度灵活,可自由安装所需包
磁盘占用占用较多磁盘空间占用极少磁盘空间

简单来说:Anaconda 是“全家桶”,Miniconda 是“最小系统”
我这里直接安装的是Anaconda,命令如下:
Anaconda安装版本这个网址可以查看更多版本信息

wgetrepo.anaconda.com/archive/Anaconda3-2023.07-2-Linux-x86_64.sh

下载完成后,运行安装脚本:

bashAnaconda3-2023.07-2-Linux-x86_64.sh

阅读并接受许可协议
安装脚本会显示许可协议,按 Enter 键滚动阅读,然后输入 yes 接受协议。
选择安装路径
安装程序会提示你选择安装路径。默认路径是 ~/anaconda3,你可以按 Enter 使用默认路径,或指定其他路径。
初始化Anaconda
安装完成后,安装程序会询问是否要初始化Anaconda。输入 yes,这样Anaconda会自动将路径添加到你的 ~/.bashrc 文件中。
Do you wish the installer to initialize Anaconda3 by running conda init? [yes|no] [no] >>> yes
激活Anaconda
为了使更改生效,你需要重新加载 ~/.bashrc 文件:

source~/.bashrc

验证安装
你可以通过以下命令验证Anaconda是否安装成功:

conda--version

再输入:

python--version

应该会显示 Anaconda 自带的 Python 版本
创建一个指定名称的空环境

conda create-n你的环境名称

创建后激活环境
环境创建完成后,需要用以下命令激活它,才能开始使用:

conda activate 你的环境名称

极少数情况下,创建环境时可能没有自动安装 pip。可以手动安装:

condainstallpip

安装CUDA

CUDA版本

echo"操作系统:$(uname-s)"echo"架构:$(uname-m)"echo"发行版:$(grep^ID=/etc/os-release|cut-d=-f2|tr-d'\"')"echo"版本:$(grep^VERSION_ID=/etc/os-release|cut-d=-f2|tr-d'\"')"

你可以直接运行以下命令汇总所有信息,这样你就可以对照官网中的选项,选择正确的 NVIDIA 驱动安装包了

wgethttps://developer.download.nvidia.com/compute/cuda/13.0.0/local_installers/cuda_13.0.0_580.65.06_linux.runsudoshcuda_13.0.0_580.65.06_linux.run

安装pytorch

pip3installtorch torchvision