1. 程式人生 > 其它 >ENVI深度學習在虛擬機器中使用TeslaT4顯示卡解決辦法

ENVI深度學習在虛擬機器中使用TeslaT4顯示卡解決辦法

ENVI深度學習在普通物理機中使用Tesla T4等顯示卡直接安裝對應顯示卡驅動即可,顯示卡驅動下載地址:https://www.nvidia.cn/Download/Find.aspx?lang=cn

在虛擬機器中使用Tesla T4顯示卡需檢查如下兩步:

1.  檢查虛擬機器對應的實體機顯示卡驅動

保證虛擬機器對應的實體機顯示卡驅動為vGPU顯示卡驅動,即虛擬工作站的GIRD驅動程式。Tesla T4顯示卡安裝grid 461.09顯示卡驅動測試成功。

顯示卡驅動下載地址:

https://pan.baidu.com/s/15Q8D0fWNHPn1B7ynU6drXw

提取碼:envi

2.  檢查虛擬機器CPU設定模式

檢查超融合雲平臺CPU模式。CPU設定成直通模式,預設模式可能為相容模式,相容模式導致一些DLL不能正確呼叫。

安裝設定好之後在ENVI Toolbox,選擇Deep Learning>Deep Learning Guide Map>Tools>Test Installation and Configuration進行檢測,環境配置成功之後顯示如下內容表示可以執行ENVI深度學習工具。

*** Testing Graphics Card ***

{

   "gpu": 0,

   "name": "Tesla T4",

   "driverVersion": "461.09",

   "computeCapability": "7.5",

   "pciBusID": "0000:00:0B.0",

   "totalMemory": 16106127360,

   "freeMemory": 15627280384

}

Good news! This graphics card can be used with ENVI Deep Learning.

*** Initializing a TensorFlow model ***

OK.

*** Creating a Label Raster ***

OK.

*** Training a TensorFlow Model ***

OK.

All tests completed.