create environments

conda create -n torch_fuck_you_311_d26191cde370489f82e544cebb585898 python=3.11.10
conda create -n torch_fuck_you_312_b7da69d4a4ff4f74919aef0207fec6d8 python=3.12.5

conda activate torch_fuck_you_311_d26191cde370489f82e544cebb585898
pip install torch==2.0.1
conda activate torch_fuck_you_312_b7da69d4a4ff4f74919aef0207fec6d8
pip install torch==2.4.1

run the fucking script with your gpu

conda activate torch_fuck_you_311_d26191cde370489f82e544cebb585898
python fuck.py

tensor(1.3328, device='cuda:0', grad_fn=<SelectBackward0>)

conda activate torch_fuck_you_312_b7da69d4a4ff4f74919aef0207fec6d8
python fuck.py

tensor(1.3323, device='cuda:0', grad_fn=<SelectBackward0>)?