TensorRT安裝小記

 docker中查看cuda和cudnn版本python 查看cuda版本 cat /usr/local/cuda/include/cudacat /usr/local/cuda/version.txt 查看cudnn版本 cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2 或 cat /usr/local/cuda/include/cudnn.h
相關文章
相關標籤/搜索