代码之家 › 专栏 › 技术社区 › Skeleton Bow

无法在WSL上启用CUDA的情况下安装火炬

nvidia gpu cuda pytorch python

0

Skeleton Bow · 技术社区 · 2 年前

在过去的几个月里,我间歇性地遇到了一个问题,即我无法在Windows 11上的WSL上安装启用CUDA的PyTorch。

我使用的是一台带有RTX 4090 GPU的Windows 11台式电脑。Windows安装已经安装并启用了WSL,我从WSL运行所有的Jupyter笔记本。

我使用了来自的以下命令 PyTorch's website 要安装火炬:

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

然而,当我在Jupyter笔记本中运行以下单元格时,我得到了否定的响应:

import torch

cuda_available = torch.cuda.is_available()
if cuda_available:
    print("CUDA is available")
else:
    print("CUDA is not available")

CUDA is not available

如何运行支持CUDA的PyTorch版本?

0 回复 | 直到 2 年前

1

SimonUnderwood 2 年前

您需要为WSL设置GPU直通。我已经完成了,但我不推荐它,因为它的设置和维护相当麻烦。

这是一套非常详细的 instructions 如果你还想试试的话。