pytorch-1.0分支编译依赖项
我的cuda版本是10.1,cuda位置在/usr/local/cuda,编译过程出现报错:
error: command '/usr/local/cuda-8.0:/usr/local/cuda-10.1/bin/nvcc' failed with exit status 1
参考这个https://github.com/jwyang/faster-rcnn.pytorch/issues/605
在命令执行前先执行:
export CUDA_HOME=/usr/local/cuda
然后再是:
python setup.py build develop
编译成功
master分支使用pytorch0.4版本的makefile修改:
https://github.com/hosang/gossipnet/issues/12
https://github.com/bgshih/aster/issues/56