Ubuntu 中 CUDA 上的 Pytorch

Ubuntu 中 CUDA 上的 Pytorch

我是初学者,有几个问题。

我的设置:

  • 显卡:GeForce GTX 1650
  • GPU 驱动程序 430.09(nvidia 主页建议)
  • Ubuntu 20.04

目标:

  • 使任意 Pytorch 版本在我的 GPU 上运行

问题:

  1. 如果建议我的 GPU 使用 430.09,我可以使用驱动程序 460 吗,还是我只能使用 430.09?
  2. 哪个 CUDA 工具包可以在我的 GPU 上运行?Pytorch 需要 9.1、9.2、10.1、10.2
  3. 在 GPU 上执行 pytorch 需要以下哪一个:Cuda-toolkit、Cuda、Nvidia 驱动程序
  4. 我是否需要仅在特定的 conda 环境中安装 Cuda-toolkit/Cuda,还是在“基础”上安装,或者在两者上都安装?

提前致谢

相关内容