簡體   English   中英

導入TensorFlow會阻止python程序運行

[英]Importing TensorFlow stops python program from running

我在安裝了CPython的Visual Studios中安裝了Python Tools。

在Visual Studios中,如果我運行以下代碼:

print("hello");
import numpy;
print("hello");

該程序運行正常,打印兩個'你好',並正常退出。

但是,如果我運行以下代碼:

print("hello");
import tensorflow;
print("hello");

該程序掛起,打印一個'你好',並拒絕繼續。

應正確安裝所有軟件包 - 使用Python交互式窗口中的TensorFlow打印正確的輸出並完美運行。

為什么程序在第二種情況下掛起?

一旦你導入tensorflow它會自動嘗試加載cuda,它會輸出如下內容:

I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcublas.so locally
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcudnn.so locally
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcufft.so locally
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcuda.so.1 locally
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcurand.so locally

所以我認為正在發生的事情是你沒有正確安裝cuda並因此而失敗。 您可以嘗試安裝不使用GPU的CPU版本,也不會加載這些庫。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM