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

gpu cuda python pytorch jupyternotebook

hkhk789123 2023-03-27 16:52:51 ‧ 1337 瀏覽

分享至

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

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

登入發表討論

直播研討會

{{ item.channelVendor }} {{ item.webinarstarted }} |

直播中

1 個回答

增廣建文

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

回應 2
分享
檢舉

hkhk789123 iT邦新手 5 級 ‧ 2023-03-31 00:21:57 檢舉

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

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

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

登入發表回應

我要發表回答

立即登入回答

參賽組數

1064 組

團體組數

40 組

累計文章數

22200 篇

完賽人數

602 人

15th鐵人賽 16th鐵人賽 13th鐵人賽 14th鐵人賽 12th鐵人賽 11th鐵人賽鐵人賽 2019鐵人賽 javascript 2018鐵人賽 python 2017鐵人賽 windows php c# windows server linux css react vue.js

IT邦幫忙