tenorflow не может найти графический процессор - PullRequest
Новая
       124

tenorflow не может найти графический процессор

0 голосов
/ 14 июля 2020

Я установил TensorFlow-GPU 2.1.0, но когда его спросили о версии, я получил ответ 1.13.1 

import tensorflow as tf
print(tf.__version__)

большая проблема заключается в том, что когда я запускаю тест GPU, ответ будет False *

Я пробовал это 

print("Num GPUs Available:",len(tf.config.experimental.list_physical_devices('GPU')))

ответ AttributeError: модуль 'tensorflow' не имеет атрибута 'config', но я запускаю этот скрипт 

from numba import vectorize, jit, cuda  
# to measure exec time 
from timeit import default_timer as timer 
# normal function to run on cpu 
def func(a):                                 
   for i in range(10000000): 
       a[i]+= 1    
# function optimized to run on gpu 
@vectorize(['float64(float64)'], target ="cuda")                         
def func2(x): 
    return x+1
# kernel to run on gpu
@cuda.jit
def func3(a, N):
    tid = cuda.grid(1)
    if tid < N:
        a[tid] += 1
if __name__=="__main__": 
    n = 10000000                            
    a = np.ones(n, dtype = np.float64) 
    for i in range(0,5):
         start = timer() 
         func(a) 
         print(i, " without GPU:", timer()-start)     
    for i in range(0,5):
         start = timer() 
         func2(a) 
         print(i, " with GPU ufunc:", timer()-start) 
    threadsperblock = 1024
    blockspergrid = (a.size + (threadsperblock - 1)) // threadsperblock
    for i in range(0,5):
         start = timer() 
         func3[blockspergrid, threadsperblock](a, n) 
         print(i, " with GPU kernel:", timer()-start)

ответы: 0 без графического процессора: 5,481482100000051 1 без графического процессора: 5,6241342000000145 2 без графического процессора: 5,62558580000001 3 без графического процессора: 5,299320899999998 4 без графического процессора: 5,424306600000023 0 с графическим процессором ufun c: 0,4764495000000011 1 с графическим процессором ufun c: 0,1611822509999 0,12550920000001042 3 с графическим процессором ufun c: 0,11633530000000292 4 с графическим процессором ufun c: 0,11252430000001823 0 с ядром графического процессора: 0.20753619999999273 1 с ядром графического процессора: 0,08865670000000136 2 с ядром графического процессора: 0,08246159999998781 3 с ядром графического процессора: 0,0824615999999878 могу ли я сделать GPU видимым для TensorFlow?

1 Ответ

0 голосов
/ 15 июля 2020

Проблема была из-за наличия на моем устройстве двух версий Cuda (10 и 11). Я удалил Cuda 10, и проблема была решена

...