fix: Fit compatibility of cuda 9.0 and pytorch 0.4.1

longcw / faster_rcnn_pytorch

Faster RCNN with PyTorch

MIT License

1.7k stars 466 forks source link

Closed J911 closed 5 years ago

J911 commented 5 years ago

test code

$ CUDA_VISIBLE_DEVICES=0,1,2,3 python train.py