安装 cudatoolkit 时缺少 Nvcc？

Question

我已经在 pytorch 上安装了 cuda

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

但是，似乎没有安装 nvcc。例如nvcc -V，如果我想使用，我会收到未找到 nvcc 的错误，我应该使用sudo apt install nvidia-cuda-toolkit. Can I do this安装它（我不想只是尝试然后发现它不起作用/弄乱了整个cuda 设置）。这是错误还是预期行为？

我使用的是 Ubuntu 18.04 并且有 cuda 10.2

Answer 1

nvidia conda 通道现已可用：nvidia/cuda和nvcc已在其中：

conda install cuda -c nvidia

Answer 2

cudatoolkit随 pytorch 安装的仅是运行时，不附带开发编译器nvcc。为了让nvcc你需要安装cudatoolkit-dev我相信可以从conda-forge渠道获得。

Answer 3

cudatoolkit使用 PyTorch 1.4安装10.1时遇到此问题。

有一个conda-forge包https://anaconda.org/conda-forge/cudatoolkit-dev。安装后，nvcc以及其他 CUDA 库将在/home/li/anaconda3/envs/<env_name>/pkgs/cuda-toolkitinbin/和lib/.

Answer 4

你可以试试

conda install -c conda-forge nvcc_linux-64

目前这应该会让你得到 10.2。Nvidia 有自己的频道 nvidia，但最新版本是 10.1。

Answer 5

cudatoolkit-dev来自的软件包conda-forge对我不起作用。HCC我使用了最新的 pytorch (v1.9.0) https://anaconda.org/HCC/cudatoolkit中的软件包。

cudatoolkit与pytorch和一起安装的命令torchvision：

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch -c hcc

安装完成后可以查看

$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Oct_23_19:24:38_PDT_2019
Cuda compilation tools, release 10.2, V10.2.89

Answer 6

检查 nvcc 是否在/usr/local/cuda-10.2/bin Run文件夹中./nvcc --version（如果该文件夹中存在）

输出看起来像这样

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Oct_23_19:24:38_PDT_2019
Cuda compilation tools, release 10.2, V10.2.89

如果是这种情况，请将文件夹添加到全局路径变量中 echo "export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}" >> ~/.profile

并使用刷新配置文件 source ~/.profile

并重新启动您的机器。