Saifeddine_Barkia:

!conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.2 -c pytorch



I don’t know, The github code didn’t provide any requirement , so I used pytorch 1.5. I will try 10.1.