File tree 1 file changed +5
-2
lines changed 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -110,8 +110,8 @@ def _check_driver():
110
110
111
111
def _check_capability ():
112
112
incorrect_binary_warn = """
113
- Found GPU%d %s which requires CUDA_VERSION >= %d for
114
- optimal performance and fast startup time , but your PyTorch was compiled
113
+ Found GPU%d %s which requires CUDA_VERSION >= %d to
114
+ work properly , but your PyTorch was compiled
115
115
with CUDA_VERSION %d. Please install the correct PyTorch binary
116
116
using instructions from https://pytorch.org
117
117
"""
@@ -126,9 +126,12 @@ def _check_capability():
126
126
for d in range (device_count ()):
127
127
capability = get_device_capability (d )
128
128
major = capability [0 ]
129
+ minor = capability [1 ]
129
130
name = get_device_name (d )
130
131
if capability == (3 , 0 ) or major < 3 :
131
132
warnings .warn (old_gpu_warn % (d , name , major , capability [1 ]))
133
+ elif CUDA_VERSION <= 9000 and major >= 7 and minor >= 5 :
134
+ warnings .warn (incorrect_binary_warn % (d , name , 10000 , CUDA_VERSION ))
132
135
133
136
134
137
def _lazy_call (callable ):
You can’t perform that action at this time.
0 commit comments