首页 > 系统相关 >WSL2-Ubuntu Pytorch深度学习开发环境搭建

WSL2-Ubuntu Pytorch深度学习开发环境搭建

时间:2024-04-10 17:02:01浏览次数:31  
标签:安装 usr WSL2 NVIDIA Pytorch cuda Ubuntu local CUDA

安装Linux发行版

删除现有Linux发行版

wsl -l -v
wsl --unregister Ubuntu

image.png
从Microsoft Store安装Linux发行版
image.png
设置用户名和密码
image.png

安装CUDA

CUDA(Compute Unified Device Architecture)是由NVIDIA推出的并行计算平台和编程模型。CUDA Toolkit是由NVIDIA提供的一套用于GPU开发的工具和库的集合。它包含了许多用于加速应用程序和算法的工具,以及用于构建和优化GPU加速软件的资源。

WSL上的CUDA用户指南 CUDA Toolkit存档

更新软件包信息并将已安装的软件包升级到它们的最新版本

sudo apt update && sudo apt upgrade

查看NVIDIA GPU相关信息

nvidia-smi

image.png
安装CUDA所需的基本依赖

sudo apt install build-essential

验证系统安装的gcc版本

gcc --version

image.png

安装CUDA Toolkit

安装适用于WSL-Ubuntu的相应版本的CUDA Toolkit,选择runfile安装类型

WSL-Ubuntu的CUDA Toolkit 12.1下载页面

image.png

wget https://developer.download.nvidia.com/compute/cuda/12.1.1/local_installers/cuda_12.1.1_530.30.02_linux.run
sudo sh cuda_12.1.1_530.30.02_linux.run

查看安装在/usr/local/cuda目录下的CUDA

ls /usr/local/cuda

image.png

**/usr/local/cuda/bin/**包含了CUDA的可执行文件,例如编译器nvcc和其他的CUDA工具。
**/usr/local/cuda/include/**包含了CUDA的头文件,这些头文件定义了CUDA库和API的接口。
**/usr/local/cuda/lib64/**包含了CUDA库文件,这些库文件是用于链接CUDA程序的。

配置环境变量

编辑Bash Shell配置文件,进行环境变量的设置

vim ~/.bashrc

按 i 键进入插入模式,在文件的末尾添加下面两行语句

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

按Ecs键返回命令模型,输入以下命令保存文件并退出vim

:wq

重新加载配置文件,使环境变量生效

source ~/.bashrc

验证CUDA的安装

查看NVIDIA CUDA编译器的版本信息

nvcc -V

image.png
运行NVIDIA CUDA Toolkit提供的示例程序

bandwidthTest是NVIDIA CUDA Toolkit提供的一个用于测试GPU内存带宽的工具。
deviceQuery是NVIDIA CUDA Toolkit提供的一个用于查询和显示GPU设备详细信息的工具。
vectorAdd是NVIDIA CUDA Toolkit提供的一个并行向量加法的示例程序。

/usr/local/cuda/extras/demo_suite/bandwidthTest
/usr/local/cuda/extras/demo_suite/deviceQuery
/usr/local/cuda/extras/demo_suite/vectorAdd

image.png

安装cuDNN

NVIDIA cuDNN(CUDA Deep Neural Network library)是由NVIDIA提供的一个用于深度学习的GPU加速库。cuDNN提供了高性能的基本操作和优化,旨在加速深度神经网络(DNN)的训练和推断过程。它是针对NVIDIA GPU架构的深度学习应用程序而设计的。

NVIDIA cuDNN安装指南

在Windows中下载适用于相应CUDA版本的cuDNN

cuDNN下载页面

image.png
将下载好的cudnn tar文件移动到Linux目录下

cp /mnt/x/xxxx/cudnn-linux-x86_64-8.9.6.50_cuda12-archive.tar.xz .

解压cuDNN包

tar -xvf cudnn-linux-x86_64-8.9.6.50_cuda12-archive.tar.xz

将cuDNN安装路径下的cuDNN头文件和库文件复制到CUDA toolkit的目录中,并赋予所有用户对 cuDNN 头文件和库文件的读取权限

sudo cp cudnn-linux-x86_64-8.9.6.50_cuda12-archive/include/cudnn*.h /usr/local/cuda/include 
sudo cp -P cudnn-linux-x86_64-8.9.6.50_cuda12-archive/lib/libcudnn* /usr/local/cuda/lib64 
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

至此CUDA和cuDNN安装完成。

安装MiniConda

Miniconda是一个轻量级的Anaconda版本,是一个开源的包管理和环境管理系统,旨在简化在不同平台上安装软件包和管理虚拟环境。Miniconda提供了一个快速、简单的方法来设置和使用Python环境。

MiniConda官方文档

下面四行命令完成了64位版本MiniConda的安装

mkdir -p ~/miniconda3
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda3/miniconda.sh
bash ~/miniconda3/miniconda.sh -b -u -p ~/miniconda3
rm -rf ~/miniconda3/miniconda.sh

将Miniconda的初始化脚本添加到Bash shell的配置文件中,以便在启动Bash shell时自动激活Miniconda环境

~/miniconda3/bin/conda init bash

配置Pytorch虚拟环境

创建一个新的Conda环境,指定Python的版本并在环境中安装pip

conda create -n pytorch211 python=3.10 pip

查看系统中已创建的Conda环境的信息

conda env list

image.png
激活指定的Conda环境

conda activate pytorch211

使用pip安装对应CUDA版本的Pytorch,同时安装Jupyter Notebook,使用清华大学的镜像源

Pytorch官网

image.png

pip install torch torchvision torchaudio notebook -i https://pypi.tuna.tsinghua.edu.cn/simple

运行Pytorch示例程序

下载Pytorch官方的示例代码

git clone https://github.com/pytorch/examples.git

运行一个使用Pytorch框架的多项式回归示例

python3 examples/regression/main.py

image.png
退出当前Conda环境

conda deactivate

标签:安装,usr,WSL2,NVIDIA,Pytorch,cuda,Ubuntu,local,CUDA
From: https://blog.csdn.net/m0_46349114/article/details/137602382

相关文章

  • 基于Anaconda安装Pytorch
    提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档文章目录前言一、安装Anaconda二、安装过程中遇见的问题1.文章3.4.2安装失败总结前言Anaconda安装Pytorch具体流程与心得。一、安装Anaconda示例:pandas是基于NumPy的一种工具,该工具是为了解决数......
  • Ubuntu22.04安装vmtools失败总结
    Ubuntu22.04安装vmtools失败总结问题按照网上安装vmtools方法,点击VMwareWorkstation菜单栏虚拟机重新安装VMtools后再运行下列代码sudo./vmware-install.pl会失败。无法与Windows进行联动。解决方案sudoapt-getupdatesudoapt-getinstallopen-vm-tools-......
  • Ubuntu Desktop 免费的文件 / 目录差异比较工具 (Beyond Compare 为收费软件)
    UbuntuDesktop免费的文件/目录差异比较工具[BeyondCompare为收费软件]1.Installation2.MeldDiffViewer3.LocktoLauncherReferencesMeld-Visualdiffandmergetoolhttps://meldmerge.org/Meldhelpsyoucomparefiles,directories,an......
  • Ubuntu安装chatGLM3-6b
    目录概述安装Anaconda3原文链接下载Anaconda3安装Anaconda3执行bash安装提示“conda:未找到命令”的解决办法anaconda3常用命令添加/更换conda清华源下载chatglm3-6b模型下载chatglm3壳子概述本文主要介绍如何在Ubuntu上安装ChatGLM3-6b模型。只有6B模型支持functi......
  • ubuntu安装AX200 wifi驱动
    第一步:先了解自己的无线网卡的品牌和型号;可以在:此电脑->管理->设备管理器->网络适配器 中找到,或者下载一个鲁大师查看第二步:下载驱动文件;例如我的无线网卡是intelAX200,我便曲intel官网下载,如: intel的linux无线网卡驱动下载链接: intel无线网卡驱动(linux)第三步:下载kernel......
  • 机器学习常见的sampling策略 附PyTorch实现
    简单的采样策略首先介绍三种简单采样策略:Instance-balancedsampling,实例平衡采样。Class-balancedsampling,类平衡采样。Square-rootsampling,平方根采样。它们可抽象为:\[p_j=\frac{n_j^q}{\sum_{i=1}^Cn_i^q},\]\(p_j\)表示从j类采样数据的概率;\(C\)表示类别数量......
  • Ubuntu下安装谷歌Chrome浏览器
    步骤1:下载Chrome谷歌浏览器,下方的指令会将Chrome安装包下载到当前目录wgethttps://dl.google.com/linux/direct/google-chrome-stable_current_amd64.deb下载完成后,安装包就存在于当前目录了,如下图:步骤2:使用管理员权限安装Chromesudoaptinstall./google-chrome-sta......
  • 记一下-ubuntu22.04安装QQ音乐
    最近用安装了桌面的ubuntu-server22.04,感觉很好用,然后突发奇想,看下装个音乐播放器但是找来找去,发现现在好多已经不开发Debian系的版本了,也就qq音乐还有就尝试安装了下,发现虽然可以安装,但是无法正常打开,打开了就闪退下面就介绍下不闪退的安装步骤下载包到qq音乐的官网下载htt......
  • vmware虚拟机重置ubuntu系统密码
    1.虚拟机重启过程中同时长按shift键,跳出下面界面 2.按键盘e键,到下面页面 3.把红框中的代码改为rwinit=/bin/bash,然后按F10键到下图 4.输入passwd用户名然后enter,一次输入两次密码,会有如下的提示,表示更新密码成功,如下图 然后重启即可,用新密码进入系统 ......
  • ubuntu16.04 wordpress建站教程
    四、wordpress搭建完成本地浏览器输入服务器IP地址,跳转至wordpress安装界面在安装界面中输入数据库密码即可完成安装本地机器输入IP地址/wp-admin进入wordpress后台登录成功......