The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37. #11
Description
NVIDIA RTX A2000 Laptop GPU with CUDA capability sm_86 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.
Has anyone else encountered this problem before? When I upgrade my PyTorch version, I can start training for PPO, but this time I get
TypeError: can't convert np.ndarray of type numpy.object_ during testing. The only supported types are: float64, float32, float16, complex64, complex128, int64, int32, int16, int8, uint8, and bool.
As @t-woodw found in 6th comment, I am also running the fix_conf.py file.
I also tried running it with 1050Ti, but the PC crashes during the training session. However, when I run the training for only 5 steps, I get successful test results, but of course, these are not sufficient.