记录一个问题,由于版本问题导致的 CUBLAS_STATUS_INVALID_VALUE
1.输入命令 python -m torch.utils.collect_env,查看版本
2.确认CUDA used to build PyTorch: 11.7 和 CUDA runtime version: 11.6.55 是否一致
(我这里不一致)
![](https://img-blog.csdnimg.cn/img_convert/dbd0d036f674bf6a8331b8a7010ada22.png)
记录一个问题,由于版本问题导致的 CUBLAS_STATUS_INVALID_VALUE
1.输入命令 python -m torch.utils.collect_env,查看版本
2.确认CUDA used to build PyTorch: 11.7 和 CUDA runtime version: 11.6.55 是否一致
(我这里不一致)