记录编译Pycharm训练代码时出现的两个问题及解决方案

1、问题描述:AttributeError: module ‘distutils‘ has no attribute ‘version‘ 

解决方案:打开报错对应的内置文件,把其中涉及到相关报错的部分:

LooseVersion = distutils.version.LooseVersion

替换成以下代码:

from distutils.version import LooseVersion

保存修改后重新运行即可解决该问题。

2、问题描述:ModuleNotFoundError: No module named 'torch._custom_ops'
解决方案:应该是torch和torchvision的版本不对应,按照环境需求配置成匹配的版本即可。笔者是安装的如下对应版本:Successfully installed torch-1.13.0 torchvision-0.14.0,其中,建议先确定torchvision,它能够自动安装对应版本的torch,注意的是,要是要使用gpu的话,需要下载和CUDA对应版本的torch和torchvision版本。

可以到如下网站寻找自己电脑CUDA版本对应的torch下载指令。

https://pytorch.org/get-started/previous-versions/

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值