摘要: 我的是torch和torchvision版本不对应,这张图是现在的torch、torchvision、python适配的版本,按照这张图的版本从新下一次torch和torchvision 阅读全文
posted @ 2022-01-22 23:39 {hunter}ZY 阅读(4233) 评论(0) 推荐(0)
摘要: 这个原因是pytorch不支持CUDA, 可以先输入 import torchprint(torch.cuda.is_available()) 如果输入为false则打开cmd,输入nvidia-smi查看cuda的版本,之后去Previous PyTorch Versions | PyTorch找 阅读全文
posted @ 2022-01-22 23:34 {hunter}ZY 阅读(4162) 评论(0) 推荐(0)
摘要: import os, sys, glob, shutil, json import cv2 from PIL import Image import numpy as np import torch from torch.utils.data.dataset import Dataset impor 阅读全文
posted @ 2022-01-22 23:28 {hunter}ZY 阅读(93) 评论(0) 推荐(0)