小编Ali*_*son的帖子

CUDA 11.8 和 Pytorch 以及 NVIDIA GeForce RTX 3050 Ti 笔记本电脑 GPU(不使用 GPU 作为计算引擎)

我正在尝试使用我的 GPU 作为 Pytorch 的计算引擎。

我在本地计算机上安装了所有带有 CUDA 11.8 的驱动程序 (522.06),但 Pytorch 无法识别我的 GPU。

我使用了不同的下载选项,最后一个:

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
Run Code Online (Sandbox Code Playgroud)

pip conda torch pytorch

7
推荐指数
1
解决办法
1万
查看次数

标签 统计

conda ×1

pip ×1

pytorch ×1

torch ×1