Question

我已经在pytorch上安装了cuda，

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

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

我正在使用Ubuntu 18.04，并具有cuda 10.2

Answer 1

您可以尝试

conda install -c conda-forge nvcc_linux-64

当前，这应该使您获得10.2。 Nvidia拥有自己的频道nvidia，但最新版本为10.1。

Answer 2

在PyTorch 1.4中安装10.1的cudatoolkit时遇到此问题。

有一个conda-forge软件包https://anaconda.org/conda-forge/cudatoolkit-dev。安装此程序后，将在nvcc和/home/li/anaconda3/envs/<env_name>/pkgs/cuda-toolkit中的bin/上提供lib/和其他CUDA库。

Answer 3

检查nvcc是否在文件夹/usr/local/cuda-10.2/bin中如果该文件夹中存在./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

然后重新启动计算机。

安装cudatoolkit时缺少NVCC吗？

3 个答案: