自由Man

cuda11.4安装什么版本pytorch?torch.cuda.is_available()结果为false?

pytorch版本与cuda版本不匹配


pytorch官方查不到cuda11.4的,据说安装cuda10.2~11.5均可以

亲测有效:

pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

发表评论:

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

Powered By Z-BlogPHP 1.5.2 Zero Theme By 爱墙纸

Copyright ZiYouMan.cn. All Rights Reserved. 蜀ICP备15004526号