Hey @ptrblck I am still getting the original error, Error in magma_getdevice_arch: MAGMA not initialized (call magma_init() first) or bad device
Although I installed using conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch