WSL2安装CUDA

WSL2安装CUDA

在WSL里面不用安装显卡驱动，显卡驱动必须要提前在Windows系统里面安装好，安装好后相应库文件会自动映射到WSL里面

1. 进入网站

2. 找到自己需要的版本，我这里是CUDA 11.7.1，点击进入详细页面

3. 依次选择$Linux\to x86\_64\to WSL-Ubuntu\to 2.0\to runfile$

   

4. 复制下面生成的命令，粘贴到shell执行

   

5. 什么都不用改，直接选择安装

   

6. 查看安装日志，按照指导添加环境变量

   

   export PATH=/usr/local/cuda-11.7/bin:$PATH
   export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64
   

7. Pytorch检查CUDA可用性，打开ipython：

   import torch
   torch.cuda.is_available()
   torch.version.cuda
   torch.tensor(12.)
   

   

图片转存中…(img-FYqbxIPl-1689521657766)]