Solución:
Me funcionó después de agregar la variable de entorno CUDA_HOME:
%%writefile setup.sh
export CUDA_HOME=/usr/local/cuda-10.1
git clone https://github.com/NVIDIA/apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./apex
!sh setup.sh
(solo quería agregar un comentario pero no tengo suficiente reputación …)
funciona para mi pero el cd en realidad no es necesario. Además, necesitaba las dos opciones globales sugeridas aquí: https://github.com/NVIDIA/apex/issues/86
%%writefile setup.sh
git clone https://github.com/NVIDIA/apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./apex
luego
!sh setup.sh
Primero, cree un archivo, por ejemplo setup.sh como sigue:
%%writefile setup.sh
git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --no-cache-dir ./
Entonces, instálalo
!sh setup.sh
¡Haz clic para puntuar esta entrada!
(Votos: 0 Promedio: 0)