cuda :
cat /usr/local/cuda/version.txt
cudnn:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
cudnn cuda version check
發表評論
所有評論
還沒有人評論，想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章
顯式調用了cuda以後，pytorch的張量還是沒有轉移帶GPU上的解決方法
在使用GPU加速計算的時候，有的時候將張量顯式調用了以後會發現其實還是在cpu上計算 打個比方： class A: def __init__(self): super(...)
400％能效提升 ARM發佈T760最快移動GPU
今天 AMR 正式發佈了兩款 Mali 系列 GPU（圖形處理器），具體型號分別爲 Mali-T720 和 Mali-T760。其中，Mali- T