[英]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.