wsl2安装cuda cudnn
时间: 2023-09-05 10:08:01 浏览: 227
要在WSL2上安装CUDA和cuDNN,你需要完成以下步骤:
1. 确保你已经在Windows主机上安装了NVIDIA显卡驱动程序,并且它是最新版本。
2. 在WSL2中安装NVIDIA的容器运行时(nvidia-docker)以及相关工具。
- 打开WSL2终端,并执行以下命令来安装nvidia-docker:
```
$ curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
$ distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
$ curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
$ sudo apt-get update && sudo apt-get install -y nvidia-docker2
$ sudo systemctl restart docker
```
3. 安装CUDA和cuDNN。
- 在WSL2终端中执行以下命令以安装CUDA和cuDNN:
```
$ sudo apt-get install nvidia-cuda-toolkit
$ sudo apt-get install libnvinfer7
```
4. 配置WSL2以使用GPU。
- 编辑WSL2配置文件 `/etc/wsl.conf`(如果文件不存在则创建),并添加以下内容:
```
[wsl2]
runtime=nvidia
```
5. 重启WSL2。
- 在WSL2终端中执行以下命令以重启WSL2:
```
$ wsl --shutdown
$ wsl
```
6. 验证CUDA和cuDNN的安装。
- 在WSL2终端中,执行以下命令以验证CUDA和cuDNN的安装:
```
$ nvidia-smi
$ nvcc --version
```
这样,你就成功在WSL2上安装了CUDA和cuDNN。请确保按照上述步骤进行操作,并根据你的需求进行相应的配置。
阅读全文