logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

两个国内的数据集网站

https://hyper.ai/datasets这个有kitti的各个任务https://www.graviti.cn/open-datasets

pytorch resnet18 train note

torch跟torchvision的有相应的对应官方推荐版本,如何确定两者之间对应的合适版本呢?有一个小技巧就是torch跟torchvision的最新版本从后往前数!因为每推出一个新版的torch,就会有一个版本的torchvision。https://download.pytorch.org/whl/torch_stable.html如图对应关系:torch和torchvision版本对应关系

python实现nvidia-smi功能

#!/usr/bin/env python# -*- coding: utf-8 -*-# @Time: 2021/11/24# @Author: linzheimport pynvmlpynvml.nvmlInit()gpunum= pynvml.nvmlDeviceGetCount()for i in range(gpunum):try:handle = (pynvml.nvmlDeviceG

#python
到底了