利用NVIDIA顯卡(CUDA)跑神經網絡

1、背景         最近在看生成對抗網絡相關的論文,須要跑神經網絡的模型,可是用筆記本跑的比較慢,因此想嘗試一下NVIDIA的顯卡。我臺式機上的顯卡是1050Ti,基本配置爲win10+keras+tensorflow-gpu。在配置的過程當中遇到了一點問題,因而記錄下來,但願能給其餘遇到相似問題的小夥伴一點幫助。我這裏利用了keras實現的GAN github源碼, 這個源碼包含了GAN及
相關文章
相關標籤/搜索