Pytorch version

Hello all,

Just thought to mention that torch==1.5.0a0+8f84ded, which is standard in the Docker image nvcr.io/nvidia/pytorch:20.03-py3, doesn’t work with the starter kit. Uninstall torch and install torch==1.3.1.

Hope this helps.

Best,
Ahmed

4 Likes