如何为 ubuntu 16.04 安装 cuda 10.2 或 cuda 11

如何为 ubuntu 16.04 安装 cuda 10.2 或 cuda 11

我想在我的 ubuntu 16.04 机器上安装 cuda 10.2 或 cuda 11.1。

我的目标是在 GPU 上运行我的 pytorch 代码,它需要这些 cuda 版本。

目前,我有一个 cuda 8,我使用以下方法进行了检查:

~$ nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2016 NVIDIA Corporation
Built on Tue_Jan_10_13:22:03_CST_2017
Cuda compilation tools, release 8.0, V8.0.61

有很多指南,但没有一个明确提到 16.04 版本。请推荐一种可行的方法。

谢谢

相关内容