我已經(jīng)在帶有 CUDA 10 和合適的 GPU 的 x86-64 Ubuntu 計(jì)算機(jī)上安裝了 tensorflow(不是從源代碼)。我的目標(biāo)是攔截(使用 LD_PRELOAD)任何 tensorflow 應(yīng)用程序的 CUDA 運(yùn)行時(shí) API。不幸的是,對于我的用例,我無法從源代碼為不是 x86-64 的目標(biāo)機(jī)器構(gòu)建 tensorflow。我能夠攔截我從用 C++ 編寫的動態(tài)加載 cuda 運(yùn)行時(shí) API 的測試程序進(jìn)行的 cudaLaunchKernel 調(diào)用,并且在第一次檢查時(shí)我認(rèn)為 python 應(yīng)該類似地動態(tài)加載相同的 cuda .so - 我很困惑,因?yàn)?LD_PRELOAD 不是適用于在啟用 cuda 的情況下運(yùn)行的正常安裝的 tensorflow 應(yīng)用程序。我希望 tensorflow 中的 cudaLaunchKernel 調(diào)用應(yīng)該被我的 LD_PRELOAD 插入庫攔截。這個(gè)對嗎?
添加回答
舉報(bào)
0/150
提交
取消