【发布时间】:2022-07-24 20:27:26
【问题描述】:
我尝试在我的电脑上运行tensorflow,但是错误消息显示如下1,我确实按照官网的确切说明通过pip安装了tensorflow,我不明白为什么会这样。有人和我有同样的问题吗?
我认为问题出在退出代码上,但我没有找到有关此代码的任何信息。
x_train shape: (50000, 32, 32, 3)
50000 train samples
10000 test samples
y_train shape: (50000, 1)
2022-01-29 23:19:00.983142: I tensorflow/core/platform/cpu_feature_guard.cc:151] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations: AVX AVX2
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
2022-01-29 23:19:01.360020: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1525] Created device /job:localhost/replica:0/task:0/device:GPU:0 with 3493 MB memory: -> device: 0, name: NVIDIA GeForce RTX 3060 Laptop GPU, pci bus id: 0000:01:00.0, compute capability: 8.6
Epoch 1/20
2022-01-29 23:19:03.467200: I tensorflow/stream_executor/cuda/cuda_dnn.cc:366] Loaded cuDNN version 8302
Process finished with exit code -1073740791 (0xC0000409)
【问题讨论】:
-
您使用的是哪个 Cuda 版本?在 Linux 上,您可以使用
nvcc —version从命令行检索它。如果您使用的是 Docker 容器,请在容器内运行命令。
标签: python tensorflow