2017-10-06 74 views
0

我有一個運行扭矩的集羣來分配作業。我想用tensorflow代碼運行一個作業,並且我遇到了tensorflow無法識別的問題。在tensorflow conda環境中運行扭矩作業

我使用anaconda在我的LDAP用戶上安裝了tensorflow,因此我可以在任何節點中輸入tensorflow環境並手動運行我的代碼。我的問題是扭矩作業在運行時無法打開conda環境,所以我得到「ImportError:No module named tensorflow」,我的代碼無法運行。因此,他們pythons代碼確實運行,但無法找到tensorflow模塊,當我在python中導入它時,因爲它不運行在tensorflow conda環境中

如何告訴節點在tensorflow conda環境中運行我的python文件?

This is how my torque job file looks

注:在這裏,我試圖運行在打開的環境中,我沒有其他版本的命令。

在此先感謝您提供任何幫助。

回答

0

我最終只需要添加我的pbs文件並將路徑導出到我的conda bin文件夾bin文件夾,因此它將使用我的conda環境python二進制文件運行python。

此外,我沒有涉及到這一點,但可能仍然與這樣做的人有關我最終需要導出我的cuda bin目錄以及。

我說什麼:

export PATH="/home/my_user/anaconda3/bin:$PATH" 
export PATH=$PATH:/usr/local/maui/bin:/usr/local/maui/sbin 
export PATH=$PATH:/usr/local/cuda-8.0/bin