CUDA HOME en la instalación de pytorch
I instalé pytorch a través de conda con cuda 7.5
conda install pytorch=0.3.0 cuda75 -c pytorch
>>> import torch
>>> torch.cuda.is_available()
True
No hice ninguna otra instalación para cuda que no sea esta, ya que parece que pytorch viene con cuda
Ahora, estoy tratando de configurar yolo2https: //github.com/longcw/yolo2-pytorc
Sin embargo, recibo un error en./make.sh
comando
este es el error
OSError: el binario nvcc no se pudo encontrar en su $ PATH. Agréguelo a su ruta o configure $ CUDAHOME
Supongo que necesito configurar CUDAHOME en mi camino, pero no puedo ubicar ningún directorio cuda que tenga nvcc binario. ¿Algún indicador sobre esto?