搜索结果
查询Tags标签: pytorch,共有 396条记录-
PyTorch 模型训练教程(一)-数据
第一章 数 据 1.1 Cifar10 转 png 下载 cifar-10-python.tar.gz 下载方式: 官网:http://www.cs.toronto.edu/~kriz/cifar.html linux命令: cd Data wget http://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz下载 cifar-10-python.tar.gz,存放到 /Data 文件夹下,…
2021/5/10 18:59:41 人评论 次浏览 -
检查cuda和pytorch版本是否匹配
目的:在已经安装pytorch和cuda的环境下,检查当前版本下pytorch和cuda版本是否兼容 在cmd输入python,运行python 输入 import torch print(torch.__version__) print(torch.cuda.is_available()) 若最终输出为false则表示pytorch和cuda版本不兼容如下图所示:若输出为tr…
2021/5/10 18:26:39 人评论 次浏览 -
自监督图像论文复现 | BYOL(pytorch)| 2020
转: 自监督图像论文复现 | BYOL(pytorch)| 2020 继续上一篇的内容,上一篇讲解了Bootstrap Your Onw Latent自监督模型的论文和结构: https://juejin.cn/post/6922347006144970760 现在我们看看如何用pytorch来实现这个结构,并且在学习的过程中加深对论文的理解。 git…
2021/5/9 10:28:26 人评论 次浏览 -
pytorch禁止/允许计算局部梯度/ torch.no_grad()
版权声明:本文为CSDN博主「Answerlzd」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。 原文链接:https://blog.csdn.net/Answer3664/article/details/99460175
2021/5/8 10:55:10 人评论 次浏览 -
Pytorch系列:(四)IO操作
首先注意pytorch中模型保存有两种格式,pth和pkl,其中,pth是pytorch默认格式,pkl还支持pickle库,不过一般如果没有特殊需求的时候,推荐使用默认pth格式保存 pytorch中有两种数据保存方法,一种是存储整个模型,一种只存储参数 方法一:存储整个模型 拼多多优惠券 htt…
2021/5/7 19:00:36 人评论 次浏览 -
[pytorch][转载]官方历史版本pytorch安装指令合集
We’d prefer you install the latest version, but old binaries and installation instructions are provided below for your convenience. COMMANDS FOR VERSIONS >= 1.0.0 v1.8.0 Conda OSX # conda conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio=…
2021/5/5 10:29:52 人评论 次浏览 -
Windows环境下配置深度强化学习环境玩Atari游戏
Windows环境下配置深度强化学习环境玩Atari游戏 1.在anaconda命令行下创建新的环境 conda create -n gym_env python=3.8.5 #可以定义自己的环境名 2.激活环境 activate gym_env 补充: 如果需要cuda加速需要安装cuda和cudnn 安装cuda: 查看自己电脑的显卡版本,去https…
2021/5/5 7:29:08 人评论 次浏览 -
pytorch镜像安装
在安装是找了很多方法如,离线安装,镜像下载等等最终还是这种方法好用,输入镜像: conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ co…
2021/5/4 18:25:46 人评论 次浏览 -
安装Pytorch的CPU版本使用pip----只需简单的四步即可完成
第一步:进入anaconda环境,建立新环境 # 创建新环境 conda create -n pytorch01(新环境名) python=3.7 # 转到新环境中 activate pytorch01第二步:进入pytorch官网查看下载命令 https://pytorch.org/get-started/locally/ 第三步:选择配置 第四步:执行命令并验证: …
2021/5/4 10:26:10 人评论 次浏览 -
Windows10+CUDA+cuDNN+Tensorflow-gpu/Pytorch安装记录
主要内容:Windows10安装CUDA,配置cuDNN,安装Tensrflow-GPU和Pytorch,Pycharm解释器配置。本电脑仅作深度学习环境配置的演示,显卡是GTX950m。本次配置过程仅供参考。 一、准备工作1、查询电脑硬件支持的CUDA 版本2、cuDNN与CUDA版本对应3、CUDA下载4、cuDNN下载5、An…
2021/5/4 7:27:35 人评论 次浏览 -
目标检测(四):SSD之Pytorch源码解读
读完 SSD 的论文内容能大致了解这一算法的核心思想和算法流程,但要将其应用到实际问题上还需要去读代码。论文给出的 SSD 源码是用 Caffe 框架实现的,但自己使用 Caffe 搭建 SSD 的环境实在是太复杂了 (师弟说他用了半个月的时间都没有弄好,到处都是坑坑坑,最后还是用…
2021/5/3 22:56:03 人评论 次浏览 -
创建pytorch/tensorflow环境后如何在pycharm中切换
创建pytorch/tensorflow环境后如何在pycharm中切换 注意本文前提是在你创建完环境之后的切换如何进行 第一步:打开settings选项,并点击右上角Add…第二步:点击conda环境,勾选existing,切换环境第三步:回到pycharm打开界面,右下角切换添加的conda环境大功告成啦!
2021/5/3 18:56:34 人评论 次浏览 -
NNLM原理及Pytorch实现
NNLM NNLM:Neural Network Language Model,神经网络语言模型。源自Bengio等人于2001年发表在NIPS上的《A Neural Probabilistic Language Model一文。 理论 模型结构任务 根据\(w_{t-n+1}...w_{t-1}\)来预测\(w_t\)是什么单词,即用\(n-1\)个单词来预测第\(n\)个单词 符…
2021/5/3 10:25:24 人评论 次浏览 -
pytorch线性回归(笔记一)
代码部分: import numpy as np import torch import matplotlib.pyplot as plt from torch import nn,optim from torch.autograd import Variablex_data=np.random.rand(100)#数组中加入在[0,1]之间均匀分布的随机样本。构建x的值 noise=np.random.normal(0,0.01,x_data…
2021/5/2 18:25:42 人评论 次浏览 -
Pytorch windows10安装教程
强烈建议安装anaconda之后再来安装这个pytorch,具体怎么安装百度搜索就知道了。温馨提示,在安装anaconda的时候记得将“添加到环境变量”(安装的时候是英文的)这一选项选上。下面假设你已经安装好anaconda了:1.第一步下载pytorch的安装包:链接: https://pan.baidu.com…
2021/5/1 7:25:46 人评论 次浏览