在pytorch中使用from apex import amp报错。pytorch安装了cuda但是没有安装nvcc

在pytorch中使用from apex import amp报错。pytorch安装了cuda但是没有安装nvcc。

报错raise RuntimeError("--cuda_ext was requested, but nvcc was not found. Are you sure your environment has nvcc available? If you're installing within a container from https://hub.docker.com/r/pytorch/pytorch, only images whose names contain 'devel' will provide nvcc.")

官方文档通过以下代码安装apex
参考：添加链接描述

$ git clone https://github.com/NVIDIA/apex
$ cd apex
$ pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

报错原因是因为没有安装nvcc。因为使用conda安装pytorch时不会通过cudatoolkit安装来安装nvcc。conda cudatoolkit只是Nvidia的子集nvidia-cuda-toolkit。
此时可以将第三行

$ pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

改成

$ python setup.py install

即可
参考：添加链接描述