What are the versions of PyTorch and CUDA？

[wmjlincy]

I have no problem training the model on the CPU environment, but I get the "Assertion index >= -size[i] && index < size[i] && "index out of bounds" failed" error on the GPU environment, my pytorch version is 1.8.1, and the cuda version is 11.1. So I wonder if it's a problem with my environment version and the author's version.