Ниже приводится код, который я использую для создания модели изображений, которые я хочу классифицировать. Я установил TensorFlow-GPU с CUDA версии 10.1. Запуск их на VS C.
import tensorflow as tf
from tensorflow.keras.datasets import cifar10
from tensorflow.keras.preprocessing.image import ImageDataGenerator
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense, Dropout, Activation, Flatten
from tensorflow.keras.layers import Conv2D, MaxPooling2D
import pickle
pickle_in = open("X.pickle","rb")
X = pickle.load(pickle_in)
pickle_in = open("y.pickle","rb")
y = pickle.load(pickle_in)
X = X/255.0
model = Sequential()
model.add(Conv2D(256, (3, 3), input_shape=X.shape[1:]))
model.add(Activation('relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Conv2D(256, (3, 3)))
model.add(Activation('relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Flatten())
model.add(Dense(32))
model.add(Dense(1))
model.add(Activation('sigmoid'))
model.compile(loss='binary_crossentropy',
optimizer='adam',
metrics=['accuracy'])
model.fit(X, y, batch_size=32, epochs=2, validation_split=0.3)
Это ошибка, которую я получаю. Есть ли способ тогда использовать только мой графический процессор вместо использования процессора и графического процессора для запуска моих программ TensorFlow?:
Failed copying input tensor from /job:localhost/replica:0/task:0/device:CPU:0 to /job:localhost/replica:0/task:0/device:GPU:0 in order to run GatherV2: Dst tensor is not initialized. [Op:GatherV2]
Вот журналы:
2020-06-20 16:32:55.699529: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_101.dll
2020-06-20 16:33:00.882825: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library nvcuda.dll
2020-06-20 16:33:00.922298: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1561] Found device 0 with properties:
pciBusID: 0000:01:00.0 name: GeForce GTX 1660 Ti computeCapability: 7.5
coreClock: 1.59GHz coreCount: 24 deviceMemorySize: 6.00GiB deviceMemoryBandwidth: 268.26GiB/s
2020-06-20 16:33:00.928892: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_101.dll
2020-06-20 16:33:00.960144: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cublas64_10.dll
2020-06-20 16:33:00.967446: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cufft64_10.dll
2020-06-20 16:33:00.979517: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library curand64_10.dll
2020-06-20 16:33:00.987231: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cusolver64_10.dll
2020-06-20 16:33:00.998423: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cusparse64_10.dll
2020-06-20 16:33:01.011126: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudnn64_7.dll
2020-06-20 16:33:01.018137: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1703] Adding visible gpu devices: 0
2020-06-20 16:33:01.031991: I tensorflow/core/platform/cpu_feature_guard.cc:143] Your CPU supports instructions that this TensorFlow
binary was not compiled to use: AVX2
2020-06-20 16:33:01.051788: I tensorflow/compiler/xla/service/service.cc:168] XLA service 0x2ed126a52c0 initialized for platform Host (this does not guarantee that XLA will be used). Devices:
2020-06-20 16:33:01.060682: I tensorflow/compiler/xla/service/service.cc:176] StreamExecutor device (0): Host, Default Version
2020-06-20 16:33:01.063495: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1561] Found device 0 with properties:
pciBusID: 0000:01:00.0 name: GeForce GTX 1660 Ti computeCapability: 7.5
coreClock: 1.59GHz coreCount: 24 deviceMemorySize: 6.00GiB deviceMemoryBandwidth: 268.26GiB/s
2020-06-20 16:33:01.078560: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_101.dll
2020-06-20 16:33:01.081907: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cublas64_10.dll
2020-06-20 16:33:01.084879: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cufft64_10.dll
2020-06-20 16:33:01.087729: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library curand64_10.dll
2020-06-20 16:33:01.098123: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cusolver64_10.dll
2020-06-20 16:33:01.101407: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cusparse64_10.dll
2020-06-20 16:33:01.114443: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudnn64_7.dll
2020-06-20 16:33:01.126877: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1703] Adding visible gpu devices: 0
2020-06-20 16:33:01.722711: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1102] Device interconnect StreamExecutor with strength 1 edge matrix:
2020-06-20 16:33:01.726195: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1108] 0
2020-06-20 16:33:01.733829: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1121] 0: N
2020-06-20 16:33:01.737104: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1247] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 4623 MB memory) -> physical GPU (device: 0, name: GeForce GTX 1660 Ti, pci bus id: 0000:01:00.0, compute capability: 7.5)
2020-06-20 16:33:01.753880: I tensorflow/compiler/xla/service/service.cc:168] XLA service 0x2ed2d417000 initialized for platform CUDA (this does not guarantee that XLA will be used). Devices:
2020-06-20 16:33:01.765729: I tensorflow/compiler/xla/service/service.cc:176] StreamExecutor device (0): GeForce GTX 1660 Ti, Compute Capability 7.5
2020-06-20 16:33:12.202959: W tensorflow/core/common_runtime/bfc_allocator.cc:434] Allocator (GPU_0_bfc) ran out of memory trying to
allocate 2.22GiB (rounded to 2384640000)
Current allocation summary follows.
2020-06-20 16:33:12.213194: I tensorflow/core/common_runtime/bfc_allocator.cc:934] BFCAllocator dump for GPU_0_bfc
2020-06-20 16:33:12.221646: I tensorflow/core/common_runtime/bfc_allocator.cc:941] Bin (256): Total Chunks: 14, Chunks in use: 14.
3.5KiB allocated for chunks. 3.5KiB in use in bin. 304B client-requested in use in bin.
2020-06-20 16:33:12.225862: I tensorflow/core/common_runtime/bfc_allocator.cc:941] Bin (512): Total Chunks: 0, Chunks in use: 0. 0B allocated for chunks. 0B in use in bin. 0B client-requested in use in bin.
2020-06-20 16:33:12.237117: I tensorflow/core/common_runtime/bfc_allocator.cc:941] Bin (1024): Total Chunks: 3, Chunks in use: 3. 3.3KiB allocated for chunks. 3.3KiB in use in bin. 3.0KiB client-requested in use in bin.
2020-06-20 16:33:12.240964: I tensorflow/core/common_runtime/bfc_allocator.cc:941] Bin (2048): Total Chunks: 0, Chunks in use: 0. 0B allocated for chunks. 0B in use in bin. 0B client-requested in use in bin.
2020-06-20 16:33:12.252326: I tensorflow/core/common_runtime/bfc_allocator.cc:941] Bin (4096): Total Chunks: 0, Chunks in use: 0. 0B allocated for chunks. 0B in use in bin. 0B client-requested in use in bin.
2020-06-20 16:33:12.264488: I tensorflow/core/common_runtime/bfc_allocator.cc:941] Bin (8192): Total Chunks: 0, Chunks in use: 0. 0B allocated for chunks. 0B in use in bin. 0B client-requested in use in bin.
2020-06-20 16:33:12.268505: I tensorflow/core/common_runtime/bfc_allocator.cc:941] Bin (16384): Total Chunks: 3, Chunks in use: 1. 76.3KiB allocated for chunks. 27.0KiB in use in bin. 27.0KiB client-requested in use in bin.
2020-06-20 16:33:12.280233: I tensorflow/core/common_runtime/bfc_allocator.cc:941] Bin (32768): Total Chunks: 0, Chunks in use: 0. 0B allocated for chunks. 0B in use in bin. 0B client-requested in use in bin.
2020-06-20 16:33:12.291124: I tensorflow/core/common_runtime/bfc_allocator.cc:941] Bin (65536): Total Chunks: 0, Chunks in use: 0. 0B allocated for chunks. 0B in use in bin. 0B client-requested in use in bin.
2020-06-20 16:33:12.295020: I tensorflow/core/common_runtime/bfc_allocator.cc:941] Bin (131072): Total Chunks: 0, Chunks in use: 0. 0B allocated for chunks. 0B in use in bin. 0B client-requested in use in bin.
2020-06-20 16:33:12.306082: I tensorflow/core/common_runtime/bfc_allocator.cc:941] Bin (262144): Total Chunks: 0, Chunks in use: 0. 0B allocated for chunks. 0B in use in bin. 0B client-requested in use in bin.
2020-06-20 16:33:12.309970: I tensorflow/core/common_runtime/bfc_allocator.cc:941] Bin (524288): Total Chunks: 0, Chunks in use: 0. 0B allocated for chunks. 0B in use in bin. 0B client-requested in use in bin.
2020-06-20 16:33:12.329149: I tensorflow/core/common_runtime/bfc_allocator.cc:941] Bin (1048576): Total Chunks: 0, Chunks in use: 0. 0B allocated for chunks. 0B in use in bin. 0B client-requested in use in bin.
2020-06-20 16:33:12.340994: I tensorflow/core/common_runtime/bfc_allocator.cc:941] Bin (2097152): Total Chunks: 1, Chunks in use: 1. 2.25MiB allocated for chunks. 2.25MiB in use in bin. 2.25MiB client-requested in use in bin.
2020-06-20 16:33:12.345317: I tensorflow/core/common_runtime/bfc_allocator.cc:941] Bin (4194304): Total Chunks: 1, Chunks in use: 0. 4.50MiB allocated for chunks. 0B in use in bin. 0B client-requested in use in bin.
2020-06-20 16:33:12.356372: I tensorflow/core/common_runtime/bfc_allocator.cc:941] Bin (8388608): Total Chunks: 0, Chunks in use: 0. 0B allocated for chunks. 0B in use in bin. 0B client-requested in use in bin.
2020-06-20 16:33:12.374974: I tensorflow/core/common_runtime/bfc_allocator.cc:941] Bin (16777216): Total Chunks: 0, Chunks in use: 0. 0B allocated for chunks. 0B in use in bin. 0B client-requested in use in bin.
2020-06-20 16:33:12.379686: I tensorflow/core/common_runtime/bfc_allocator.cc:941] Bin (33554432): Total Chunks: 0, Chunks in use: 0. 0B allocated for chunks. 0B in use in bin. 0B client-requested in use in bin.
2020-06-20 16:33:12.391123: I tensorflow/core/common_runtime/bfc_allocator.cc:941] Bin (67108864): Total Chunks: 0, Chunks in use: 0. 0B allocated for chunks. 0B in use in bin. 0B client-requested in use in bin.
2020-06-20 16:33:12.403099: I tensorflow/core/common_runtime/bfc_allocator.cc:941] Bin (134217728): Total Chunks: 0, Chunks in use: 0. 0B allocated for chunks. 0B in use in bin. 0B client-requested in use in bin.
2020-06-20 16:33:12.407256: I tensorflow/core/common_runtime/bfc_allocator.cc:941] Bin (268435456): Total Chunks: 3, Chunks in use: 1. 4.51GiB allocated for chunks. 915.75MiB in use in bin. 915.75MiB client-requested in use in bin.
2020-06-20 16:33:12.419033: I tensorflow/core/common_runtime/bfc_allocator.cc:957] Bin for 2.22GiB was 256.00MiB, Chunk State:
2020-06-20 16:33:12.421277: I tensorflow/core/common_runtime/bfc_allocator.cc:963] Size: 1.79GiB | Requested Size: 915.75MiB | in_use: 0 | bin_num: 20, prev: Size: 2.25MiB | Requested Size: 2.25MiB | in_use: 1 | bin_num: -1, next: Size: 915.75MiB | Requested Size: 915.75MiB | in_use: 1 | bin_num: -1
2020-06-20 16:33:12.434709: I tensorflow/core/common_runtime/bfc_allocator.cc:963] Size: 1.83GiB | Requested Size: 0B | in_use: 0 | bin_num: 20, prev: Size: 915.75MiB | Requested Size: 915.75MiB | in_use: 1 | bin_num: -1
2020-06-20 16:33:12.447599: I tensorflow/core/common_runtime/bfc_allocator.cc:970] Next region of size 4847737088
2020-06-20 16:33:12.450046: I tensorflow/core/common_runtime/bfc_allocator.cc:990] InUse at 70ae00000 of size 1280 next 1
2020-06-20 16:33:12.460040: I tensorflow/core/common_runtime/bfc_allocator.cc:990] InUse at 70ae00500 of size 256 next 5
2020-06-20 16:33:12.462539: I tensorflow/core/common_runtime/bfc_allocator.cc:990] InUse at 70ae00600 of size 1024 next 8
2020-06-20 16:33:12.472602: I tensorflow/core/common_runtime/bfc_allocator.cc:990] InUse at 70ae00a00 of size 256 next 10
2020-06-20 16:33:12.474863: I tensorflow/core/common_runtime/bfc_allocator.cc:990] InUse at 70ae00b00 of size 256 next 11
2020-06-20 16:33:12.477063: I tensorflow/core/common_runtime/bfc_allocator.cc:990] InUse at 70ae00c00 of size 1024 next 9
2020-06-20 16:33:12.504675: I tensorflow/core/common_runtime/bfc_allocator.cc:990] InUse at 70ae01000 of size 256 next 15
2020-06-20 16:33:12.509815: I tensorflow/core/common_runtime/bfc_allocator.cc:990] InUse at 70ae01100 of size 256 next 16
2020-06-20 16:33:12.511951: I tensorflow/core/common_runtime/bfc_allocator.cc:990] InUse at 70ae01200 of size 256 next 14
2020-06-20 16:33:12.522050: I tensorflow/core/common_runtime/bfc_allocator.cc:990] InUse at 70ae01300 of size 256 next 19
2020-06-20 16:33:12.536646: I tensorflow/core/common_runtime/bfc_allocator.cc:990] InUse at 70ae01400 of size 256 next 20
2020-06-20 16:33:12.539029: I tensorflow/core/common_runtime/bfc_allocator.cc:990] InUse at 70ae01500 of size 256 next 21
2020-06-20 16:33:12.550313: I tensorflow/core/common_runtime/bfc_allocator.cc:990] InUse at 70ae01600 of size 256 next 22
2020-06-20 16:33:12.552654: I tensorflow/core/common_runtime/bfc_allocator.cc:990] InUse at 70ae01700 of size 256 next 23
2020-06-20 16:33:12.562774: I tensorflow/core/common_runtime/bfc_allocator.cc:990] InUse at 70ae01800 of size 256 next 24
2020-06-20 16:33:12.565098: I tensorflow/core/common_runtime/bfc_allocator.cc:990] Free at 70ae01900 of size 22528 next 2
2020-06-20 16:33:12.567281: I tensorflow/core/common_runtime/bfc_allocator.cc:990] InUse at 70ae07100 of size 256 next 3
2020-06-20 16:33:12.578171: I tensorflow/core/common_runtime/bfc_allocator.cc:990] InUse at 70ae07200 of size 256 next 4
2020-06-20 16:33:12.580584: I tensorflow/core/common_runtime/bfc_allocator.cc:990] Free at 70ae07300 of size 27904 next 6
2020-06-20 16:33:12.591431: I tensorflow/core/common_runtime/bfc_allocator.cc:990] InUse at 70ae0e000 of size 27648 next 7
2020-06-20 16:33:12.593765: I tensorflow/core/common_runtime/bfc_allocator.cc:990] Free at 70ae14c00 of size 4718592 next 13
2020-06-20 16:33:12.596024: I tensorflow/core/common_runtime/bfc_allocator.cc:990] InUse at 70b294c00 of size 2359296 next 12
2020-06-20 16:33:12.606133: I tensorflow/core/common_runtime/bfc_allocator.cc:990] Free at 70b4d4c00 of size 1920466944 next 18
2020-06-20 16:33:12.608381: I tensorflow/core/common_runtime/bfc_allocator.cc:990] InUse at 77dc54c00 of size 960233472 next 17
2020-06-20 16:33:12.619338: I tensorflow/core/common_runtime/bfc_allocator.cc:990] Free at 7b7014c00 of size 1959873792 next 18446744073709551615
2020-06-20 16:33:12.622295: I tensorflow/core/common_runtime/bfc_allocator.cc:995] Summary of in-use Chunks by size:
2020-06-20 16:33:12.632770: I tensorflow/core/common_runtime/bfc_allocator.cc:998] 14 Chunks of size 256 totalling 3.5KiB
2020-06-20 16:33:12.635109: I tensorflow/core/common_runtime/bfc_allocator.cc:998] 2 Chunks of size 1024 totalling 2.0KiB
2020-06-20 16:33:12.637392: I tensorflow/core/common_runtime/bfc_allocator.cc:998] 1 Chunks of size 1280 totalling 1.3KiB
2020-06-20 16:33:12.646671: I tensorflow/core/common_runtime/bfc_allocator.cc:998] 1 Chunks of size 27648 totalling 27.0KiB
2020-06-20 16:33:12.648955: I tensorflow/core/common_runtime/bfc_allocator.cc:998] 1 Chunks of size 2359296 totalling 2.25MiB
2020-06-20 16:33:12.651221: I tensorflow/core/common_runtime/bfc_allocator.cc:998] 1 Chunks of size 960233472 totalling 915.75MiB
2020-06-20 16:33:12.660794: I tensorflow/core/common_runtime/bfc_allocator.cc:1002] Sum Total of in-use chunks: 918.03MiB
2020-06-20 16:33:12.663097: I tensorflow/core/common_runtime/bfc_allocator.cc:1004] total_region_allocated_bytes_: 4847737088 memory_limit_: 4847737241 available bytes: 153 curr_region_allocation_bytes_: 9695474688
2020-06-20 16:33:12.674876: I tensorflow/core/common_runtime/bfc_allocator.cc:1010] Stats:
Limit: 4847737241
InUse: 962627328
MaxInUse: 2883092480
NumAllocs: 52
MaxAllocSize: 960233472
2020-06-20 16:33:12.687934: W tensorflow/core/common_runtime/bfc_allocator.cc:439] *______________________________________*********************________________________________________