pytorch 环境配置(Linux)

在深度学习领域,Windows 系统虽普及度高,但在实际应用中存在诸多局限。一方面,部分深度学习框架的 API 仅支持 Linux 系统。另一方面,当深度学习与车辆控制、机器人控制等领域相结合时,ROS(机器人操作系统)的使用几乎不可或缺,而 ROS 在 Linux 系统上的兼容性和稳定性表现更为出色。故针对在Linux中安装深度学习环境以及Linux中常用的工具
本博客主要针对安装了 Windows 和 Linux 双系统的电脑展开,至于在电脑中安装双系统的教程,可参考 Windows11+Linux(Ubuntu22.04)双系统最简安装详细避坑版

一、Nvidia 显卡驱动安装
  1. 安装步骤

    1. 禁用 Nouveau 驱动
    • 创建禁用 Nouveau 的配置文件:
    sudo nano /etc/modprobe.d/blacklist-nouveau.conf
    
    • 在文件中添加以下内容:
    blacklist nouveau
    options nouveau modeset=0
    
    • 更新 initramfs:
    sudo update-initramfs -u
    
    • 重启系统:
    sudo reboot
    
  2. 安装Nvidia驱动

    • 赋予文件执行权限,我在官网下载的驱动文件为NVIDIA-Linux-x86_64-550.144.03.run ,请在安装的时候根据自己下载的驱动文件执行对应的指令:
    chmod +x NVIDIA-Linux-x86_64-550.144.03.run 
    
    • 执行安装命令
    sudo ./NVIDIA-Linux-x86_64-550.144.03.run 
    
    • 根据下述步骤一路安装即可,其中会出现一些警告,直接选择Continue即可

在这里插入图片描述
在这里插入图片描述

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

  • 在安装完成后重启系统后,在终端中输入 nvidia-smi,如果有结果则证明安装成功
nvidia-smi
二、 Anaconda3 安装
  • 安装地址:https://www.anaconda.com/download/success

  • 安装介绍:https://blog.csdn.net/mieshizhishou/article/details/140269614

  • 初始安装目录为_root/anaconda3_ 。但是在 linux 中每个账户都有一个目录,同时每个账户都有自己的环境变量,故如果安装在_root_ 主目录下。当使用其他普通用户时便无法使用 anaconda3,故需要更改安装目录为当前所使用账户的主目录
    在这里插入图片描述

    • 因为我当前所使用的账号是_zjut_ ,同时对应的主目录为_/home/zjut_ ,故修改地址为 /home/zjut/anaconda3
      在这里插入图片描述

    • 这部分选择 yes 后会自动初始化_Anaconda_ 。在初始化_Anaconda_ 后输入 source /home/zjut/.bashrc 即可加载配置
      在这里插入图片描述

    • 最后关闭终端再打开即可使用_Anaconda_

三、pytorch 环境配置
  • 下载地址:https://pytorch.org/

  • 创建一个新的环境用于安装 pytorch

    • conda create

      • 这是 Conda 的命令,用于创建一个新的虚拟环境。
    • -n pytorch

      • -n--name 的缩写,用于指定虚拟环境的名称。
      • pytorch 是虚拟环境的名称,你可以根据需要自定义。
    • python=3.12

      • 指定虚拟环境中安装的 Python 版本为 3.12。
      conda create -n pytorch python=3.12
      
  • 查看CUDA版本

    nvidia-smi
    
四、Vscode 安装配置
sudo dpkg -i 文件名.deb
五、linux基础知识
  1. 账号与目录的关系
  • 每个用户账号都有一个主目录
    • 在 Linux 中,每个用户账号都有一个主目录(Home Directory),通常位于 /home/用户名 下。
    • 例如:
      • 用户 zjut 的主目录是 /home/zjut
      • 用户 root 的主目录是 /root
  • 目录并不直接代表账号
    • 目录是文件系统的一部分,用于存储文件和子目录。
    • 账号是系统中的一个实体,用于管理用户权限和资源。
  1. 环境变量与 .bashrc
  • 每个用户都有自己的环境变量
    • 环境变量是用户会话的一部分,用于配置 Shell 的行为。
    • 每个用户的环境变量是独立的,可以通过 .bashrc 文件进行设置。
  • .bashrc** 文件**:
    • .bashrc 是 Bash Shell 的配置文件,位于用户的主目录下(如 /home/zjut/.bashrc/root/.bashrc)。
    • 每次用户登录或打开新的终端时,都会加载 .bashrc 文件中的配置。
    • 如果是登陆的zjut则会加载/home/zjut/.bashrc,而不会加载root/.bashrc
六、搜狗输入法安装
Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐