初学python遇到PyTorch AttributeError: module ‘torch._C‘ has no attribute ‘_cuda_setDevice‘的问题

初学者在使用PyTorch时遇到AttributeError,原因是torch版本与CUDA不匹配。解决方案包括检查CUDA版本,确保torch版本与CUDA和Python版本相兼容,从官网安装torch库,以及在安装时使用清华源以确保稳定性。如果无法使用GPU,可以将代码中的GPU属性设为-1。
摘要由CSDN通过智能技术生成

记录原因:

初次使用AI,遇到PyTorch AttributeError: module ‘torch._C‘ has no attribute ‘_cuda_setDevice‘的问题

使用场景:初学者记录

初次使用网上的代码,重要的是要引入第三方库以及路径,我对代码一直有兴趣,但后来发现,短时间变牛皮是不可能的,但我从大佬的人工智能程序里找到了乐趣。

安装库

使用requirements.txt安装库,在Anaconda输入。

python -m pip install -r G:\requirements.txt

注:requirement.txt文件在G盘,-r是读取文件,-m是在以脚本模式启动,详情:
https://www.cnblogs.com/xueweihan/p/5118222.html


torch版本不正确

一般情况下使用torch库需要CUDA的支持,需要先查看CUDA版本(torch版本要低于CUDA版本尽量一样),并且torch版本号要与Python版本相同。如果CUDA版本小于9.2࿰

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值