Install pytorch with CUDA 11

14

Yeah, I had to install pytorch 1.7.0 which worked fine.

Command I used:

conda install pytorch==1.7.0 cudatoolkit=11.0 -c pytorch