Background
最近有个项目需要跑深度学习,懒狗懒得装双系统,于是尝试在wsl中解决。
搜了一些资料后发现win10的开发者预览版中支持wsl2使用cuda,就开始折腾了。
安装步骤
1. 安装Win10预览版系统
依次进入设置->更新与安全->Windows预览体验计划,将预览版本切换到Dev Channel
,然后检查更新,安装到最新版本。(因为该版本是内测版本,可能有一些不可知的Bug,所以在安装更新前请务必备份好您的个人资料。)
当前版本为21332。
2. 安装WSL2
参考官方文档安装WSL2。
3. 安装CUDA on WSL 驱动程序
在驱动下载链接中下载安装支持cuda on wsl2的驱动程序,当前的版本号为470.05
4. 在WSL2中安装CUDA
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
sudo sh -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/cuda.list'
sudo apt-get update
sudo apt-get install -y cuda-toolkit-11-0
需要注意的是,上方两个链接中的1804
需要替换为您所使用的版本号,如2004
等。
5. 测试是否安装成功
在WSL中运行CUDA Sample,如/usr/local/cuda/samples/1_Utilities/deviceQuery
的deviceQuery,可以得到类似的输出:
至此,就可以在WSL中愉快地炼丹了!