WSL2 中大模型环境一步到位！（wsl --update 不好使/wsl2安装/python环境)

embedded/2024/11/14 12:50:37/

前言

需要在WSL上玩深度学习，需要以下几个条件

win11，最好更新到最新版本
电脑上有显卡，Nvdia
windows上安装显卡驱动及CUDA和CuDNN
安装WSL2
WLS2安装好Ubuntu20.04（其他版本大概率有很多坑）

在进行前在你的 cmd 执行下面命令确认下你的显卡驱动是正确的：

nvidia-smi

在这里插入图片描述

WSL 安装

wsl --install -d Ubuntu-20.04

如果安装报错：

在这里插入图片描述

直接改注册表：

win+R输入regedit进入注册表编辑器

找到以下文件夹路径

计算机\HKEY_LOCAL_MACHINE\SOFTWARE\Classes\Drive\shell\WSL

计算机\HKEY_LOCAL_MACHINE\SOFTWARE\Classes\Directory\background\shell\WSL

计算机\HKEY_LOCAL_MACHINE\SOFTWARE\Classes\Directory\shell\WSL

将文件夹名称右键重命名为

计算机\HKEY_LOCAL_MACHINE\SOFTWARE\Classes\Drive\shell\WSL_bak

计算机\HKEY_LOCAL_MACHINE\SOFTWARE\Classes\Directory\background\shell\WSL_bak

计算机\HKEY_LOCAL_MACHINE\SOFTWARE\Classes\Directory\shell\WSL_bak

再重新启动WSL可以正常使用了

安装完后再你的 ubuntu 命令行中就可以执行 nvidia-smi 了。但是一定要是使用的 WSL2。下面 Docker 文章中回复区写了 WSL2 的官方文档，可以去看看怎么设置。

Docker 环境

见我另一篇文章 window11 设置 ubuntu2204 至最佳体验(安装/右键菜单/root用户/docker

Python 环境

别名更改

每次都用 python3 xx 不方便，所以直接改为使用 python

查看python版本

 python3 --version

创建一个连接就行了

 sudo ln -s /usr/bin/python3.8 /usr/bin/python

在这里插入图片描述

pip 安装&源更换

先安装 pip

apt update

apt list --upgradable

sudo apt install python3-pip

上面的命令执行过程有提示就按Y。

pip 源更换

不更换的话安装东西很慢. 执行下面某一个源的命令就行。建议清华源。

清华源
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple阿里源
pip config set global.index-url https://mirrors.aliyun.com/pypi/simple/腾讯源
pip config set global.index-url http://mirrors.cloud.tencent.com/pypi/simple豆瓣源
pip config set global.index-url http://pypi.douban.com/simple/换回默认源
pip config unset global.index-url