当前位置: 首页 > article >正文

windows在wsl ubuntu环境中启用cuda加速AI推理和训练

如果在windows系统下运行AI模型和训练模型遇到各种环境问题,建议通过wsl在ubuntu环境行跑AI。

操作系统版本要求window10及以上

一、WSL安装

wsl是windows系统自带的程序,可参考官方文档操作https://learn.microsoft.com/zh-cn/windows/wsl/install,或按下面步骤操作:
① 在管理员模式下打开 PowerShell 或 Windows 命令提示符。
② 安装wsl命令:wsl --install,然后重启计算机
③ 切换成wsl2:wsl --set-default-version 2
④ 更新wsl版本:wsl --update
⑤ 输入 wsl --install -d Ubuntu 命令安装Ubuntu虚拟环境,然后根据提示操作
⑥ 安装完后可以在开始或搜索中找到Ubuntu菜单,打开后根据提示设置账号密码,然后就可以进入到Ubuntu系统命令行
在这里插入图片描述

二、安装CUDA环境

参考nvidia官方指南:https://docs.nvidia.com/cuda/wsl-user-guide/index.html#getting-started-with-cuda-on-wsl
可按下面命令逐步操作:

sudo apt-key del 7fa2af80
wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-wsl-ubuntu.pin
sudo mv cuda-wsl-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda-repo-wsl-ubuntu-12-4-local_12.4.0-1_amd64.deb
sudo dpkg -i cuda-repo-wsl-ubuntu-12-4-local_12.4.0-1_amd64.deb
sudo cp /var/cuda-repo-wsl-ubuntu-12-4-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda-toolkit-12-4

安装完成后,通过nvidia-smi命令查看显卡驱动信息,显示如下:
在这里插入图片描述
可以看到,显卡RTX 4090被正常识别,至此cuda环境安装完成。


http://www.kler.cn/a/551106.html

相关文章:

  • 【DeepSeek 行业赋能】从金融到医疗:探索 DeepSeek 在垂直领域的无限潜力
  • UART(一)——UART基础
  • 云贝餐饮连锁V3独立版全开源+vue源码
  • 领克汽车工厂探秘品牌体验方案拆解
  • 蓝桥杯备赛1-2合法日期
  • 一文掌握Docker基础
  • Cloud: aws:network: limit 含有pps这种限制
  • 渲染相机设置 pyrender cameralib
  • AWS上Amazon Redshift用Zoominfo API验证公司基本信息数据正确性检查设计方案
  • 从二维到三维3D工业相机如何改变机器视觉检测
  • vue3项目实践心得-寻找未被使用的最小编号
  • Maven指南-从入门到精通
  • 动手学深度学习11.7. AdaGrad算法-笔记练习(PyTorch)
  • Linux 下 VIM 编辑器学习记录:从基础到进阶(上)
  • Java 设计模式之命令模式
  • 深入解析 Webpack 的构建流程:一场高级技术冒险
  • 基于豆瓣2025电影数据可视化分析系统的设计与实现
  • c# sqlite 批量生成insert语句的函数
  • matlab欠驱动船舶模型预测控制
  • 【项目实践06】【Retrofit2 框架的使用】