最近在学习Triton,它需要linux的环境来运行nvidia-docker,由于不想装双系统,只能通过安装windows子系统(WSL)来解决,这里我介绍一下我的安装过程以及踩的坑。
第一步:安装nvidia驱动,这里需要安装最新的window版的驱动(不要安装linux的),我之前安装过旧版的驱动不兼容WSL,需要重新安装,可以去网站下载Official Drivers | NVIDIA
根据自己电脑GPU的配置选择驱动版本 ,我这里是GTX 1080
下载好了直接点击安装包进行安装
第二步:安装WSL2 ,挺简单的我这里不细说了,不会的可以看看B站视频(我安装的是ubuntu18.04)
Window10开发环境搭建(1) | 详细演示WSL2的安装_哔哩哔哩_bilibili,
这里需要注意wsl2系统必须是20145及以上版本,通过 ctrl+windows图标 键入 winver 查看版本,
内核必须是4.19.121+版本,通过uname -r 命令查看,如果版本太低,在windows设置里面选择更新
点击高级选项,打开更新开关
第三步:安装cuda-toolkit(我安装的是最新版的11.6,由于网络配置问题,我选择从阿里云镜像下载)
进入ubuntu18.04,首先执行:
apt-key adv --fetch-keys https://mirrors.aliyun.com/nvidia-cuda/ubuntu1804/x86_64/7fa2af80.pub
有可能会出现 gpg: WARNING: unable to fetch URI https://mirrors.aliyun.com/nvidia-cuda/ubuntu1804/x86_64/7fa2af80.pub: No data
出现这个问题一般是没有添加ubuntu的公钥F60F4B3D7FA2AF80,可以执行添加
sudo gpg --keyserver keyserver.ubuntu.com --recv F60F4B3D7FA2AF80
sudo gpg --export --armor F60F4B3D7FA2AF80 | sudo apt-key add -
sudo apt-get update
然后接着执行
sudo sh -c 'echo "deb https://mirrors.aliyun.com/nvidia-cuda/ubuntu1804/x86_64 /" > /etc/apt/sourc
es.list.d/cuda.list'
再更新一下apt
sudo apt-get update
最后安装cuda-toolkit-11-6
sudo apt-get install -y cuda-toolkit-11-6
时间较长,耐心等待一下,安装完成之后再检测一下,11.6版的cuda-samples需要自己到github上下载:https://github.com/nvidia/cuda-samples
git clone https://github.com/NVIDIA/cuda-samples.git
下载好之后执行
sudo cd <sample_dir>
sudo make
cd 1_Utilities/deviceQuery
./deviceQuery
出现下面这个结果,说明安装成功了
注意:如果出现 helper_cuda.h:779 code=35(cudaErrorInsufficientDriver)错误
说明驱动装错了,最好安装最新的驱动
接下来就是安装nvidia-docker