- 博客(8)
- 收藏
- 关注
原创 torch训练一些小细节
torch训练一些小细节一、模板:二、加入梯度剪切一、模板:经典训练模板:for i, (image, label) in enumerate(train_loader): # 1. input output pred = model(image) loss = criterion(pred, label) # 2. backward optimizer.zero_grad() # reset gradient loss.backward()
2022-02-17 16:19:25
2437
原创 python os模块基础操作
python os模块基础操作一、根据路径创建文件夹一、根据路径创建文件夹给定一个特定文件的完整路径,判断这个路径是否完整存在,若不存在则创建这个路径if not os.path.exists(_path): #path不要以‘/’或‘\\’结尾 os.makedirs(_path, exist_ok=True) #递归创建文件按夹...
2021-11-15 11:17:28
669
原创 Linux X86自带python环境下opencv+torch+torchvision安装
Linux X86自带python环境下opencv+torch+torchvision安装1.pip安装2.opencv安装3.torch安装4.opencv安装1.pip安装安装pipsudo apt install python-pip后续利用pip安装完库,pip list查看库存在,但是import 时又显示库不存在,所以又安装了pip3sudo apt install python3-pip后续利用pip3安装opencv时出现以下错误:Command “python setup
2021-09-28 16:49:49
576
原创 list 删除多个指定位置的元素
list 删除多个指定位置的元素 a=[1,2,3,4,5,6] index=[1,3] for i,idx in enumerate(index): idx=idx-i a.pop(idx)
2021-09-09 14:45:08
1835
原创 ARM64/TX2配置anaconda+torch+opencv
ARM64/TX2配置anaconda+torch+opencv1.anacondaAnaconda要安装arm64架构的,下载链接: link.安装过程与正常的anaconda安装相同。2.torchtorch要也要安装arm64架构的,下载链接:link.通过pip进行安装:安装过程出现错误:这是因为.whl文件的命名不规范。找到pap425tags.py文件,查看命名规范,对下载好的.whl文件名进行修改。我是将torch-1.6.0-cp36-cp36m-linux_aarc
2021-06-23 16:11:56
1261
原创 ubuntu18.04安装CUDA10.0与cuDNN7.6.5
ubuntu18.04安装CUDA10.0与cuDNN7.6.5一、安装CUDA10.01.检查版本和类型2.禁用nouveau驱动3.安装驱动4.安装CUDA5.安装cuDNN一、安装CUDA10.01.检查版本和类型首先查看自己的系统版本,命令如下:可以看到我的系统是Ubuntu18.04,在此基础上查看适合本系统的NVIDIA显卡型号和推荐的驱动程序模型。返回以下结果:可以看出,推荐的驱动型号为 nvidia-driver-455。安装该驱动:2.禁用nouveau驱动在终端输
2020-12-30 15:51:35
1037
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人