0

我最近从 windows 切换到 ubuntu。我在这里安装了 miniconda,使用 python 3.8.5 创建了一个虚拟环境,激活了该环境并运行

conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch 

我跑了torch.cuda.is_available(),它说 True,但是当我运行 which CUDA/Cuda/cuda 或 whereis CUDA/Cuda/cuda 时,我什么也得不到。nvcc 也不起作用。我想我可能需要更新 bash 文件。有人可以指导我吗?

另外,我是否需要安装 cuDNN 才能使用 CUDA 处理我的 ML 模型,还是不需要。

4

0 回答 0