Ubuntu搭建Pytorch，就这一篇就够了

您所在的位置：网站首页 › ubuntu怎么安装python37 › Ubuntu搭建Pytorch，就这一篇就够了

Ubuntu搭建Pytorch，就这一篇就够了

2023-12-24 21:43| 来源: 网络整理| 查看: 265

在实践的过程中，对于 Ubuntu 搭建 Pytorch 环境做了多次修改，步骤基本上都是无误的可以放心食用，如果文章对你有帮助，记得点赞呀👍

第一步：配置镜像源，安装必要环境

我在这篇文章中详细写了如何配置清华源更换镜像

第二步：安装anaconda 1 进入官网

https://www.anaconda.com/

2 下载linux的sh版

3 在对应位置通过命令输入 sh 文件名.sh 4 选择 YES

5 选择 NO （我们要自定义配置环境）

第三步：安装完成Anconda后配置环境 1 编辑环境变量 vim ~/.bashrc 2 添加内容

这里的自己对应的路径名，一般默认安装会有提示，一般都是home/用户名

export PATH="/自己对应的路径名/anaconda3/bin:$PATH" 3 激活修改的内容 source ~/.bashrc 4 测试 conda

若没有显示not fond 则表示 anaconda安装成功

第四步：通过Anaconda创建pytorch空间，用于pytorch

前提知要：

使用anaconda安装环境后，本地默认环境成为base环境

自己创建的环境通过 conda activate 空间名字（环境名字）进入

退出自己创建的环境到默认环境：conda deactivate

1 创建空间,命名为pytorch conda create -n pytorch 2 进入pytorch空间 conda activate pytorch

ps，若在执行命令的时候出现一下问题：

执行

source activate

第五步：在base环境中安装cuda 1 安装 nvidia-cuda-toolkit 工具 sudo apt-get install nvidia-cuda-toolkit 2 检查系统推荐显卡驱动，记录下recommend选项 sudo ubuntu-drivers devices

2022.12.14：更新记录，记录下recommend前面的数字，比如这里470，到时候更新驱动选择该数字短的驱动

3 添加驱动源 sudo add-apt-repository ppa:graphics-drivers/ppa sudo apt-get update 4 在Ubuntu系统中找到软件和更新选择驱动（有更新记录，看下面文字！）

2022.12.14：更新记录，这里不要选择第一个！可能会出现黑屏现象。这里选择其它同名的。

等待更新完毕后再reboot

reboot 5 重启后的测试 nvidia-smi

记录 cuda 版本

6 下载cuda

官网

https://developer.nvidia.com/cuda-toolkit-archive

我的是11.4

选择11.4的哪个版本都可以，接下去按照如下选择去下载

注意：linux的版本要确保正确

7 安装cuda(执行自己上面红框的内容) wget https://developer.download.nvidia.com/compute/cuda/11.4.4/local_installers/cuda_11.4.4_470.82.01_linux.runsudo

2022.12.14：更新记录，如果sh无反应或者需要权限：

chmod 755 sudo sh sh cuda_11.4.4_470.82.01_linux.run

2022.12.14：更新记录，这里选择是Continue，截图的时候有问题

8 环境配置 sudo gedit ~/.bashrc

末尾加入

路径中的 cuda 注意需要是自己实际安装的 cuda 文件夹，可以进入/usr/local目录下确认

export PATH=/usr/local/cuda/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH 9 测试 source ~/.bashrc nvcc -V cd ~/NVIDIA_CUDA-11.4_Samples/1_Utilities/bandwidthTest/ make ./bandwidthTest

第六步：在pytorch环境中安装 pytorch环境 1 进入官网

https://pytorch.org/

2 安装

选择pytorch版本选择好对应的系统选择使用什么安装，conda里可以使用pip安装的选择语言：python 选择cuda版本：我的版本是11.4 在这里插入图片描述

复制命令

3 通过命令进入pytorch环境 source activate conda activate pytorch 4 执行自己官网复制的命令 pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

done～撒花 💐

番外

删除conda环境

conda uninstall -n 名称 --all

【本文地址】

公司简介

联系我们