我已经安装了cuda和pytorch
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
然而,看起来nvcc并没有随它一起安装。例如,如果我想使用nvcc -V
,我得到的错误是找不到nvcc,并且我应该用sudo apt install nvidia-cuda-toolkit.
安装它。我可以这样做吗(我不想只是尝试然后发现它不工作/弄乱整个cuda设置)。这是一个bug还是预期的行为?
我使用的是Ubuntu 18.04和cuda 10.2
发布于 2020-04-16 22:55:53
在使用PyTorch 1.4安装10.1的cudatoolkit
时遇到了这个问题。
有一个conda-forge
包https://anaconda.org/conda-forge/cudatoolkit-dev
。安装后,nvcc
以及其他CUDA库将在/home/li/anaconda3/envs/<env_name>/pkgs/cuda-toolkit
的bin/
和lib/
中可用。
发布于 2020-02-19 07:06:17
你可以试试
conda install -c conda-forge nvcc_linux-64
目前,这应该会让你得到10.2。Nvidia有自己的频道nvidia,但最新版本是10.1。
发布于 2021-08-26 10:15:25
来自conda-forge
的cudatoolkit-dev
包对我不起作用。我在最新的pytorch (v1.9.0) https://anaconda.org/HCC/cudatoolkit中使用了来自HCC
的包。
将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
https://stackoverflow.com/questions/56470424
复制相似问题