tensorRT+windows

1、找到對應的cuda版本的tensorRT,下載後解壓 將lib文件夾中的dll拷貝到安裝cuda的lib文件夾中 2、在python環境中conda建立虛擬環境、安裝tensorflow-gpu 3、pip安裝tensorRT文件夾中graphsurgeon和uff 4、下載mnist數據 5、打開\TensorRT-6.0.1.5\data\mnist運行python generate_pg
相關文章
相關標籤/搜索
本站公眾號
   歡迎關注本站公眾號,獲取更多信息