WSL搭建CUDA环境

原创已于 2022-04-10 20:44:28 修改 · 5.3k 阅读

22 ·

CC 4.0 BY-SA版权

文章标签：

#linux #ubuntu #CUDA #WSL

于 2022-04-10 20:34:34 首次发布

Environment Configuration 专栏收录该内容

3 篇文章

订阅专栏

WSL搭建CUDA环境

文章目录

WSL搭建CUDA环境

打开虚拟机功能

打开控制面板—>程序与功能—>启用或关闭Windows功能，勾选以下功能

Hyper-V
适用于Linux的Windows子系统
虚拟机平台

重启

安装cuda-wsl驱动

驱动下载地址：https://developer.nvidia.com/cuda/wsl
根据你自己显卡型号选择对应驱动

下载并安装

安装WSL2

到应用商店安装WSL，我这里选择Ubuntu-18.04进行安装

WSL安装完成后查看WSL版本

wsl -l -v

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-EgN5IvWt-1649593363145)(C:\Users\onel\AppData\Roaming\Typora\typora-user-images\image-20220410195736898.png)]$
如果是WSL1，需要升级到WSL2

下载Linux 内核更新包并安装，下载地址
将Ubuntu18.04 设置为WSL2
```
wsl --set-version Ubuntu-18.04 2
```
查看wsl版本，看是否变成版本2
```
wsl -l -v
```

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

如果缺少依赖报错就安装下面依赖，安装前记得先按下面步骤换源，否则下载速度贼慢

sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev

WSL更换源，加速软件下载

sudo cp /etc/apt/sources.list /etc/apt/sources.list.bak

sudo chmod 777 /etc/apt/sources.list

sudo vim /etc/apt/sources.list

在vim命令模式下按d，删除里面内容

按i切换到编辑模式将下面内容粘贴进去

按esc按键回到命令模式按:wq保存文件

阿里源：

deb http://mirrors.aliyun.com/ubuntu/ bionic main restricted universe multiverse
deb-src http://mirrors.aliyun.com/ubuntu/ bionic main restricted universe multiverse
deb http://mirrors.aliyun.com/ubuntu/ bionic-security main restricted universe multiverse
deb-src http://mirrors.aliyun.com/ubuntu/ bionic-security main restricted universe multiverse
deb http://mirrors.aliyun.com/ubuntu/ bionic-updates main restricted universe multiverse
deb-src http://mirrors.aliyun.com/ubuntu/ bionic-updates main restricted universe multiverse
deb http://mirrors.aliyun.com/ubuntu/ bionic-proposed main restricted universe multiverse
deb-src http://mirrors.aliyun.com/ubuntu/ bionic-proposed main restricted universe multiverse
deb http://mirrors.aliyun.com/ubuntu/ bionic-backports main restricted universe multiverse
deb-src http://mirrors.aliyun.com/ubuntu/ bionic-backports main restricted universe multiverse

更新一下包

sudo apt update

安装Miniconda

# 切换到家目录
cd ~
# 下载miniconda
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
# 修改运行权限
chmod 777 Miniconda3-latest-Linux-x86_64.sh
# 运行安装脚本
./Miniconda3-latest-Linux-x86_64.sh

conda init

搭建Pytorch环境验证

创建环境

conda create --name torch python=3.8

激活环境

conda activate torch

安装pytorch，Pytorch官网地址

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

验证cuda环境

conda activate torch

python

import torch
torch.cuda.is_available()

恭喜你，环境搭建成功

彩蛋

WSL2可以直接使用图形界面打开软件，使用VScode打开当前目录

code .