文章目錄
- 前言
- 嘗試方法
- 一、嘗試方法一
- 二、嘗試方法二
- 總結
前言
之前用服務器跑的時候,發現是可以跑的。但當有其他人一同使用的時候,就會拋出:No CUDA GPUs are available,這個時候我嘗試了以下兩種方式解決,后面終于解決了
嘗試方法
一、嘗試方法一
import os
os.environ['CUDA_VISIBLE_DEVICES'] = "0"
當時能運行一段時間但之后顯示被占用,于是用nvidia-smi查看內存占用情況,發現這個時間段用的人比較多
二、嘗試方法二
import os
os.environ['CUDA_VISIBLE_DEVICES'] = "1"
把0改成1再次運行,就能正常運行了,看了看程序運行時間,跑了8個小時才跑完執行腳本
總結
除了使用os.environ[‘CUDA_VISIBLE_DEVICES’] = “1”,在此基礎上也可以嘗試調小代碼中的patience、epochs值試下