
pytorch框架
pytorch框架是一种常用框架
消灭BUG鸭
交流分享,共同进步
展开
-
ImportError: cannot import name ‘XXXXX‘ from ‘XX‘解决方案
一、报错:File "./xxx/demo/demo.py", line 23, in <module> from predictor import FeatureExtractionDemoImportError: cannot import name 'FeatureExtractionDemo' from 'predictor' 二、原因分析:1、自己的python文件与导入的包重名了,导致引入路径错乱。【主要是 .py 文件路径的问题,当在同一文件下,存在子文件内有同样原创 2021-11-02 11:22:46 · 62684 阅读 · 3 评论 -
NameError: name ‘nn‘ is not defined解决记录
NameError: name ‘nn’ is not defined1、报错界面ameError: name 'nn' is not defined2、分析原因使用Pytorch框架,需引入torch和torch.nn(或只是nn),这是两个主要的 PyTorch 包:import torchimport torch.nn as nn需要包含这两行,因为如果您只设置第二行,如果torch未导入包,则可能无法正常工作。可以**help(torch.nn)**确认这一点。3、解决方案原创 2021-10-18 09:47:34 · 14058 阅读 · 5 评论 -
AssertionError: CUDA unavailable, invalid device 0 requested
1、查看报错Traceback (most recent call last): File "train.py", line 651, in <module> device = select_device(opt.device, batch_size=opt.batch_size) File "/home/camila/test/xxx/utils/torch_utils.py", line 71, in select_device assert torch.cuda.原创 2021-06-24 09:46:28 · 62852 阅读 · 29 评论 -
虚拟环境已安装pytorch,而在pycharm中显示ModuleNotFoundError: No module named ‘torch‘的解决办法
虚拟环境已安装pytorch,而在pycharm中显示ModuleNotFoundError: No module named 'torch’的解决办法在pycharm中运行程序,明明系统和虚拟环境都已经安装pytorch,但是在pycharm显示ModuleNotFoundError: No module named ‘torch’。如下图:通过File >> setting >> Python Interpreter >> “+” >> 搜索t原创 2021-03-17 21:31:19 · 6778 阅读 · 4 评论 -
torch.cuda.is_available()返回False的解决办法
torch.cuda.is_available()返回False的解决办法之前显卡驱动和cuda都可以正常使用,但是今天运行程序时发现使用了CPU,没有使用GPU,运行的时候会报错说 GPU驱动没有运行之类的错误,然后使用CPU进行了训练(直观感觉就是慢了很多)。1.首先查看cuda 版本。cat /usr/local/cuda/version.txt// 输出正常: CUDA Version 11.0.2282.查看是否能使用GPU。结果为Falsepython3import t原创 2021-03-30 09:49:32 · 1809 阅读 · 1 评论 -
2021-03-10解决Bug记录
1.RuntimeError: Found 0 images in subfolders of: ./data/aaa原因分析: 文件夹aaa下没有子文件夹报错解决问题: 在文件夹aaa建立符合要求的自文件夹,符合数据格式。(类似这种)2.UserWarning: Detected call of lr_scheduler.step() before optimizer.step(). In PyTorch 1.1.0 and later, you should call them in the原创 2021-03-10 17:05:04 · 400 阅读 · 0 评论