WSL2 中大模型环境一步到位!(wsl --update 不好使/wsl2安装/python环境)
前言
需要在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