记录原因:

初次使用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,且python版本大于3.8(找不到与CUDA相同的版本),需要重新安装CUDA。
https://www.codenong.com/cs107054886/(CUDA安装教程)
https://developer.nvidia.com/cuda-toolkit-archive(CUDA官网安装)
https://download.pytorch.org/whl/torch_stable.html(torch镜像源)
https://pytorch.org/get-started/locally/(torch官网安装)
‘’’

安装torch库必须从官网安装

由于torch、CUDA、python版本要相互配合,所以要下载相应所需要的版本。

CUDA是CPU版本

修改代码GPU的属性为-1(我没有效果)或者是重新安装CUDA(我使用此效果成功)

使用清华源

python pip install -i https://pypi.tuna.tsinghua.edu.cn/simple/

-i https://pypi.tuna.tsinghua.edu.cn/simple/ 指从清华源镜像安装库。
从官网安装的话,由于网络不稳定,经常安装失败所以用清华源或者豆瓣源的话,好处就是又快又稳

总结

大部分出现的错误是库与库的兼容性问题,版本越高越容易产生这个问题,可以学习去建立虚拟环境,即可解决问题。我当初玩的程序是大神Jack的教程。

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐