使用未启用 CUDA 的 GPU 进行深度学习

Question

我的笔记本电脑上有 Nvidia GeForce GTX 1650。它没有启用cuda。我可以将此 GPU 与 TensorFlow 和 pytorch 一起使用吗？除了 CUDA 之外我还可以使用什么吗？

我尝试安装不同版本的CUDA，然后使用相关命令导入pytorch。但 GPU 未在可用设备中显示。

Answer 1

我正在将 TensorFlow 和 PyTorch 与 Nvidia GeForce GTX GPU 一起使用，即使它不支持 CUDA。

只需找到以下 TensorFlow 的说明即可

https://www.tensorflow.org/install/pip#amd_gpu

还有一个在 AMD GPU 上运行的解决方案，称为 ROCm PyTorch。

https://rocm.docs.amd.com/projects/install-on-linux/en/latest/how-to/3rd-party/pytorch-install.html