後來出現跑程式No CUDA GPUs are available的錯誤,爬文看到版上有人說改成os.environ['CUDA_VISIBLE_DEVICES'] = "0" 即可 (原作者是使用 os.environ['CUDA_VISIBLE_DEVICES'] = [2,3] ),但這樣子會造成out of memory的錯誤
但我發現os.environ['CUDA_VISIBLE_DEVICES']這行意思好像選擇GPU要用哪個,但是我GPU0是intel的內顯,GPU1才是NVIDIA獨立顯卡,改成1還是不行
我在工作管理員看效能GPU1 是0%感覺沒在運作,但我有做1.更改图形处理器下方的选项,更改为【高性能NVIDIA处理器】2.用python檢查版本(下圖),看起來沒啥錯誤QQ
程式指令:
import os
os.environ['CUDA_VISIBLE_DEVICES']= '1'
os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"
以下是我的錯誤訊息:
我很努力爬文還是找不到解法,拜託各位前輩們指點迷津~謝謝
用torch.cuda.device_count()看下pytorch能识别出来几张显卡,如果就一张,那么直接设置成“0”就是了,
这个不管核显的,估计你的核显也只是英特尔的而不会是英伟达的,而如果你没有经过一些特殊的处理,cuda就是只能调用英伟达的显卡,英特尔的核显是用不了cuda的