WSL2安装CUDA

作者：笔触狂放9 | 2024-02-22 04:31:12

踩

wsl2安装cuda

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

进入网站
找到自己需要的版本，我这里是CUDA 11.7.1，点击进入详细页面
依次选择 $Linux\rightarrow x86\_64\rightarrow WSL-Ubuntu\rightarrow 2.0\rightarrow runfile$
复制下面生成的命令，粘贴到shell执行
什么都不用改，直接选择安装

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

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

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

import torch
torch.cuda.is_available()
torch.version.cuda
torch.tensor(12.)
1
2
3
4

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

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/笔触狂放9/article/detail/128511