- PYTORCH=/path/to/pytorch # usually
~/anaconda3/envs/CenterNet/lib/python3.6/site-packages/
for pytorch v0.4.0
sed -i “1194s/torch.backends.cudnn.enabled/False/g” ${PYTORCH}/torch/nn/functional.py
for pytorch v0.4.1
sed -i “1254s/torch.backends.cudnn.enabled/False/g” ${PYTORCH}/torch/nn/functional.py
sed -i “1254s/torch.backends.cudnn.enabled/False/g” ~/anaconda3/envs/centernet/lib/python3.6/site-packages/torch/nn/functional.py