在google云上安装nvcc

Question

我正尝试在GCP (Google Cloud Platform)上使用pytorch。代码库要求在常规CUDA之上使用nvcc。

我已经尝试了几种方法，但到目前为止都没有奏效。

1) NVIDIA (https://ngc.nvidia.com/catalog/containers/nvidia:pytorch)部署模板。模板已配置cuda，但未配置nvcc。它建议运行apt install nvidia-cuda-toolkit来安装nvcc，这确实适用于nvcc安装，但破坏了整个cuda。

2)按照说明从头开始安装nvidia的东西，例如https://medium.com/@jayden.chua/quick-install-cuda-on-google-cloud-compute-6c85447f86a1。也不起作用：

The following packages have unmet dependencies:
 cuda : Depends: cuda-10-1 (>= 10.1.243) but it is not going to be installed
E: Unable to correct problems, you have held broken packages.

以前有人在GCP上成功安装过CUDA + nvcc吗？

感谢并致以最美好的祝福

Answer 1

首先，请确保您备份了google cloud实例。

然后，尝试通过调用以下命令卸载当前安装的cuda、nvidia包：

apt-get --purge remove "*nvidia-cuda-*"
apt-get --purge remove "*cublas*"
apt-get --purge remove "*nvidia*"
apt-get --purge remove "*cuda*"
apt-get autoremove

然后，删除此处的文件夹：

rm -rf /usr/local/cuda-*/
rm -rf /var/cuda-repo-*/

根据您的操作系统名称和版本，您可以查看可用的NVIDIA CUDA Toolkit installers。如果要为Linux Ubuntu 18.04 x86_64安装最新的CUDA Toolkit 10.1 update2 version 10.1.243，可以使用：

# CUDA Toolkit 10.1 update2 https://developer.nvidia.com/cuda-10.1-download-archive-update2
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda-repo-ubuntu1804-10-1-local-10.1.243-418.87.00_1.0-1_amd64.deb
dpkg -i cuda-repo-ubuntu1804-10-1-local-10.1.243-418.87.00_1.0-1_amd64.deb
apt-key add /var/cuda-repo-10-1-local-10.1.243-418.87.00/7fa2af80.pub
apt-get update
apt-get -y install cuda=10.1.243-1
rm -f "cuda-repo-ubuntu1804-10-1-local-10.1.243-418.87.00_1.0-1_amd64.deb"
apt autoremove

最后，使用以下命令验证安装：

nvcc --version

您应该会看到如下所示的输出：

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:07:16_PDT_2019
Cuda compilation tools, release 10.1, V10.1.243