iT邦幫忙

0

在jupyter notebook 運行CycleGan 的程式錯誤, 無法捕捉GPU 設置

  • 分享至 

  • xImage

小弟嘗試運行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

但依然出這個錯誤
https://ithelp.ithome.com.tw/upload/images/20230327/20150584Ig4PxLbrr9.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

希望有高人可指點迷津,真的查了很久也找不到方法才上網求救。

圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

1 個回答

0
增廣建文
iT邦研究生 5 級 ‧ 2023-03-27 23:38:34

感覺是一個很常見的問題

https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix/issues?q=module+%27torch._C%27+has+no+attribute+%27_cuda_setDevice%27

挑了幾個部分人回應說有效的

  • code裡面加torch._C._cuda_setDevice(-1)
  • 只用CPU跑--gpu_ids -1

如果很想用GPU跑建議直上nvidia docker

https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix/blob/master/docs/docker.md

都解決不了, 感覺可能要重新弄個新環境..

增廣建文 iT邦研究生 5 級 ‧ 2023-04-01 19:27:24 檢舉

直接用nvidia-docker吧 就不用浪費時間搞環境

我要發表回答

立即登入回答