用PyTorch写的python脚本,程序执行结束后,不自动终止

本文探讨了在使用PyTorch编写Python脚本时遇到的问题,焦点在于loss.backward()导致程序无法正常退出。提供了可能的解决方案,包括升级包和环境调整,并分享了相关经验与建议。
部署运行你感兴趣的模型镜像

问题描述:

用 PyTorch 写的一个python 脚本,脚本内容执行结束后,程序不能正常结束,按Ctrl + C也没用!!(虽然这个bug似乎对实验结果没什么影响,但是,很影响心情有没有,我居然不能控制自己写的代码……)

究竟是哪里出了问题??

经过筛查,发现问题是下面这句代码引起的:
loss.backward()

如何解决?

直接删了肯定不行,模型还跑不跑了,一通搜索之后,我似乎找到了解决办法 :网友解答

附录:

看完上面那个链接内容之后,你大概率会笑喷,或许这就是作为程序员的一种快乐吧,虽然没有解决问题~~
如果非要解决这个bug,可以尝试换系统或升级PyTorch包,若成功解决了上述问题,欢迎交流。
附上我的执行环境
OS: Win7
python 3.7.1
PyTorch: 1.5.1+cpu

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch 2.5

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

PowerShell 7 环境已加载 (版本: 7.5.2) PS C:\Users\Administrator\Desktop> cd E:\PyTorch_Build\pytorch PS E:\PyTorch_Build\pytorch> python -m venv rtx5070_env PS E:\PyTorch_Build\pytorch> .\rtx5070_env\Scripts\activate (rtx5070_env) PS E:\PyTorch_Build\pytorch> # 创建虚拟环境 (rtx5070_env) PS E:\PyTorch_Build\pytorch> python -m venv rtx5070_env Error: [Errno 13] Permission denied: 'E:\\PyTorch_Build\\pytorch\\rtx5070_env\\Scripts\\python.exe' (rtx5070_env) PS E:\PyTorch_Build\pytorch> .\rtx5070_env\Scripts\activate (rtx5070_env) PS E:\PyTorch_Build\pytorch> (rtx5070_env) PS E:\PyTorch_Build\pytorch> # 设置执行策略 (rtx5070_env) PS E:\PyTorch_Build\pytorch> Set-ExecutionPolicy RemoteSigned -Scope Process -Force (rtx5070_env) PS E:\PyTorch_Build\pytorch> (rtx5070_env) PS E:\PyTorch_Build\pytorch> # 创建所有脚本文件 (rtx5070_env) PS E:\PyTorch_Build\pytorch> # [粘贴上面的四个脚本创建命令] (rtx5070_env) PS E:\PyTorch_Build\pytorch> (rtx5070_env) PS E:\PyTorch_Build\pytorch> # 运行完整构建流程 (rtx5070_env) PS E:\PyTorch_Build\pytorch> .\full_build.ps1 (rtx5070_env) PS E:\PyTorch_Build\pytorch> (rtx5070_env) PS E:\PyTorch_Build\pytorch> # 验证CUDA编译 (rtx5070_env) PS E:\PyTorch_Build\pytorch> .\compile_cuda_test.ps1 ParserError: E:\PyTorch_Build\pytorch\compile_cuda_test.ps1:56 Line | 56 | '@ | Set-Content compile_cuda_test.ps1 | ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ | The string is missing the terminator: '. (rtx5070_env) PS E:\PyTorch_Build\pytorch> (rtx5070_env) PS E:\PyTorch_Build\pytorch> # 验证PyTorch安装 (rtx5070_env) PS E:\PyTorch_Build\pytorch> python -c "import torch; print(f'PyTorch版本: {torch.__version__}'); print(f'CUDA可用: {torch.cuda.is_available()}')" PyTorch版本: 2.8.0+cpu CUDA可用: False (rtx5070_env) PS E:\PyTorch_Build\pytorch>
最新发布
09-03
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值