Optimal

[This link](https://stackoverflow.com/questions/57814535/assertionerror-torch-not-compiled-with-cuda-enabled-in-spite-upgrading-to-cud) may help you. Note: You need to change the 'cu000' part to match your CUDA version. For example, change it to 'cu121' if you're using CUDA 12.1 or...