2024年5月25日
Tensors of the same index must be on the same device and the same dtype except `step` tensors that can be CPU and float32 notwithstanding
摘要： 避免使用 torch.set_default_dtype(torch.float64) 可以尝试采用 model. Double 或者 model.to(torch. Double) m = torch().to(device).to(torch.float64) 参考： Tensors of th 阅读全文
posted @ 2024-05-25 17:14 听风不成泣 阅读(80) 评论(0) 推荐(0) 编辑
UserWarning: Applied workaround for CuDNN issue, install nvrtc.so
摘要： 这似乎是 CUDA 库的问题pytorch 2.0.x+cu118 我采用的方法是 cd /home/user/anaconda3/env/torch/lib sudo ln -sfn libnvrtc.so.11.8.89 libnvrtc.so 参考： Convolutions are brok 阅读全文
posted @ 2024-05-25 17:09 听风不成泣 阅读(362) 评论(0) 推荐(0) 编辑