tensorflow在AMD上用ROCm框架實現GPU訓練(親測有效)

衆所周知,a卡對於tensorflow gpu的支持很小,要想使用的話通常都得配n卡的電腦。而我只是感興趣一個項目,想要跑出來一個模型,若是不用gpu加速,可能得跑一個月,因此不得不尋找在a卡上的解決方案。好在,終於找到了一個解決方案,雖然只是取巧,速度也不如正常n卡跑的快,但已經很不錯了。html 個人配置: 顯卡:RX 550 所用系統:虛擬機ubuntu 18 注:若是是虛擬機的話,最好分配
相關文章
相關標籤/搜索