pytorch学习
Let'sCode
搬砖,搬砖,再搬砖
展开
-
Ubuntu下用conda 安装 PyTorch速度慢
配置清华源:gedit ~/.condarc将下面内容复制粘贴:channels: - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/ - https://mirr原创 2021-01-27 11:36:07 · 511 阅读 · 0 评论 -
如何安装coco以及导入coco_utils和coco_eval?
今天跑pytorch目标检测教程时发现自己没有安装coco,于是按照如下步骤安装cocoubuntu@ubuntu:~$ git clone https://github.com/cocodataset/cocoapi.gitubuntu@ubuntu:~$ cd cocoapi/PythonAPIubuntu@ubuntu:~$ python setup.py build_ext --inplaceubuntu@ubuntu:~$ sudo python setup.py install原创 2021-01-20 17:20:59 · 1943 阅读 · 0 评论 -
pytorch 下载对应版本的libtorch
1.查看pytorch版本终端输入import torchprint(torch.__version__)2.查看cuda版本方法1:cat /usr/local/cuda/version.txt方法2:nvcc -V3.点击这里选择合适的libtorch版本请注意,在Linux上提供了两种类型的libtorch二进制文件:一种是用GCC pre-cx11 ABI编译的,另一种是用GCC-cx11 ABI编译的,应该根据系统使用的GCC ABI进行选择。参考文档https:/原创 2021-01-13 14:01:10 · 2332 阅读 · 0 评论 -
从pytorch代码入手,一文搞懂卷积和池化之后图片尺寸改变
在阅读代码的时候,总是会被神经网络的复杂结构搞昏头脑,尤其是经过一系列的卷积池化操作之后,完全不知道输入输出的尺寸。本文将从pytorch代码入手,推出每一步操作之后变量的尺寸。预备知识:一、卷积操作假设输入图像尺寸为wxh,卷积核尺寸为fxf,步幅为s,Padding为p,那么经过卷积层后输出的图像尺寸为:w_new = ( w + 2p -f ) / 2 + 1h_new = ( h + 2p -f ) / 2 + 1二、池化操作假设输入图像尺寸为wxh,池化核尺寸为fxf,步幅为s,那原创 2021-01-07 14:55:11 · 2464 阅读 · 1 评论 -
unexpected EOF, expected 65478 more bytes. The file might be corrupted.问题解决方案
原因分析下载pytorch的预训练模型权重时,网络不稳定,模型只下载了30%左右,再次运行,报unexpected EOF, expected xxxxxx more bytes. The file might be corrupted。这告诉我们这个文件可能已经损坏了,把损坏的权重文件删除即可。解决方案解决步骤:1.删除损坏的权重文件2.重新下载pytorch的权重文件,一般在路径\home\用户名.cache\torch文件夹下面,注意.cache是一个隐藏文件夹,在home目录下,使用 l原创 2021-01-04 14:25:31 · 7664 阅读 · 4 评论