_cuda_setDevice？？？

Open qing-tian-meng-ying opened this issue 1 year ago • 1 comments

{AttributeError}module 'torch._C' has no attribute '_cuda_setDevice' 1710422217232 明明是官方指定的torch版本，但跑起来就有问题了？

Mar 14 '24 13:03 qing-tian-meng-ying

你可以卸掉以前的，用conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia试一下，我也是遇到这个问题，用这个后解决了

Mar 15 '24 06:03 daiguoxi