Unverified Commit a7b52a68 authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Don't error on cuda minor version mismatch (#7726)

parent 41f9c1e6
...@@ -74,9 +74,9 @@ def _check_cuda_version(): ...@@ -74,9 +74,9 @@ def _check_cuda_version():
t_version = torch_version_cuda.split(".") t_version = torch_version_cuda.split(".")
t_major = int(t_version[0]) t_major = int(t_version[0])
t_minor = int(t_version[1]) t_minor = int(t_version[1])
if t_major != tv_major or t_minor != tv_minor: if t_major != tv_major:
raise RuntimeError( raise RuntimeError(
"Detected that PyTorch and torchvision were compiled with different CUDA versions. " "Detected that PyTorch and torchvision were compiled with different CUDA major versions. "
f"PyTorch has CUDA Version={t_major}.{t_minor} and torchvision has " f"PyTorch has CUDA Version={t_major}.{t_minor} and torchvision has "
f"CUDA Version={tv_major}.{tv_minor}. " f"CUDA Version={tv_major}.{tv_minor}. "
"Please reinstall the torchvision that matches your PyTorch install." "Please reinstall the torchvision that matches your PyTorch install."
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment