Closed isaacmg closed 6 years ago
Currently when trying to run a PyTorch model weights saved in CUDA we get the following error. raise AssertionError("Torch not compiled with CUDA enabled")
raise AssertionError("Torch not compiled with CUDA enabled")
Our code should be able to load the model regardless if it was trained on CPU or GPU.
This was fixed. Reference
Currently when trying to run a PyTorch model weights saved in CUDA we get the following error.
raise AssertionError("Torch not compiled with CUDA enabled")
Our code should be able to load the model regardless if it was trained on CPU or GPU.