首页 > 解决方案 > 试图让 tensorflow 2 与 WSL 2 一起工作时遇到问题

问题描述

使用 TensorFlow 后端。2021-01-09 17:01:36.009126:W tensorflow/stream_executor/platform/default/dso_loader.cc:55] 无法加载动态库“libcuda.so.1”;dlerror:libcuda.so.1:无法打开共享对象文件:没有这样的文件或目录 2021-01-09 17:01:36.009169:E tensorflow/stream_executor/cuda/cuda_driver.cc:313] 调用 cuInit 失败:未知错误(303) 2021-01-09 17:01:36.009212: I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:156] 内核驱动程序似乎没有在此主机上运行 (DESKTOP-91JP32H):/proc/driver/nvidia /版本不存在 2021-01-09 17:01:36.010807: I tensorflow/core/platform/cpu_feature_guard.cc:143] 您的 CPU 支持未编译此 TensorFlow 二进制文件以使用的指令:AVX2 FMA 2021-01-09 17:01:36.058753:我 tensorflow/core/platform/profile_utils/cpu_utils.cc:102] CPU 频率:3792000000 Hz 2021-01-09 17:01:36.059597: I tensorflow/compiler/xla/service/service.cc:168] XLA 服务 0x7fbb78000b20 已为平台主机初始化(这不保证将使用 XLA)。设备:2021-01-09 17:01:36.059635:I tensorflow/compiler/xla/service/service.cc:176] StreamExecutor 设备(0):主机,默认版本

这是完整的错误。我从 WSL 升级到 WSL2,因为从我在 WSL 1 中阅读的内容来看,它无法找到您的 GPU。我经历了这些步骤https://developer.nvidia.com/cuda/wsl没有运气。

from keras.layers import Dropout, Flatten, Dense, Activation, BatchNormalization
from keras.layers.convolutional import MaxPooling2D, Conv2D
from keras.models import Sequential
from keras.preprocessing.image import ImageDataGenerator
from keras.callbacks import ModelCheckpoint

...

model = Sequential()
model.add(Conv2D(nb_filters1, [conv1_size, conv1_size], input_shape=(img_height, img_width , 3)))```

标签: pythonkerastensorflow2.0windows-subsystem-for-linuxwsl-2

解决方案


推荐阅读