小弟嘗試運行CycleGan的代碼時遇到困擾我幾天的程序錯誤
AttributeError: module 'torch._C' has no attribute '_cuda_setDevice'
CycleGan的來源我純粹直接使用人家的code運行
來源:https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix
但不知為何出現錯誤。
以下是我電腦的配置版本:
顯示卡:4070Ti
CPU:13600kf
Pytorch:1.13.1
Python:3.9.16
CUDA:11.6
CuDnn:8!
已測試能捕捉GPU, Kernal 的環境設置也調成pytorch的自定義虛擬環境
https://ithelp.ithome.com.tw/upload/images/20230327/20150584mBa4oBaMb4.png
但依然出這個錯誤
上網查了很久有找到有人說會不會是pytorch CPU版本的問題, 但我下載的是pytorch舊版本的1.13.1 cuda 11.6版本, 如寫明cuda照計應不會使用cpu吧?
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.6 -c pytorch -c nvidia
希望有高人可指點迷津,真的查了很久也找不到方法才上網求救。
感覺是一個很常見的問題
挑了幾個部分人回應說有效的
torch._C._cuda_setDevice(-1)
--gpu_ids -1
如果很想用GPU跑建議直上nvidia docker
https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix/blob/master/docs/docker.md