搜索结果
查询Tags标签： torch，共有 290条记录-
2021-11-13
Pytorch学习（1）—— 构建网络模型的四种操作 Pytorch学习（1）—— 构建网络模型的四种操作第一种方法第二种方法第三种方法第四种方法Pytorch学习（1）—— 构建网络模型的四种操作 利用pytorch来构建网络模型有很多种方法，以下简单列出其中四种。 假设构建一个网络模…
2021/11/14 6:13:11 人评论 次浏览 -
ubuntu16.04系统pytorch安装
首先创建虚拟环境 conda create -n torch python=3.6(指定默认python为3.6)激活torch source activate torch安装torch Pytorch官方网站：https://pytorch.org/ 安装时需要注意torch和torchvision的对应版本，如下所示 根据自己的系统和python已经cuda版本进行选择安装 官…
2021/11/12 7:13:21 人评论 次浏览 -
ubuntu16.04系统pytorch安装
首先创建虚拟环境 conda create -n torch python=3.6(指定默认python为3.6)激活torch source activate torch安装torch Pytorch官方网站：https://pytorch.org/ 安装时需要注意torch和torchvision的对应版本，如下所示 根据自己的系统和python已经cuda版本进行选择安装 官…
2021/11/12 7:13:21 人评论 次浏览 -
是否需要self.loss.cuda()
转自：https://discuss.pytorch.org/t/why-do-we-need-to-do-loss-cuda-when-we-we-have-already-done-model-cuda/91023/5 https://discuss.pytorch.org/t/move-the-loss-function-to-gpu/20060 1.问题 有的模型将损失函数也调用了cuda():if torch.cuda.is_available():n…
2021/11/11 23:12:11 人评论 次浏览 -
是否需要self.loss.cuda()
转自：https://discuss.pytorch.org/t/why-do-we-need-to-do-loss-cuda-when-we-we-have-already-done-model-cuda/91023/5 https://discuss.pytorch.org/t/move-the-loss-function-to-gpu/20060 1.问题 有的模型将损失函数也调用了cuda():if torch.cuda.is_available():n…
2021/11/11 23:12:11 人评论 次浏览 -
08 线性回归 + 基础优化算法
线性回归的从零开始实现 包括数据流水线、模型、损失函数和小批量随机梯度下降优化器 # 如果没有d2l包，可以通过下面的语句安装 # !pip install -U d2l# %matplotlib inline可以在Ipython编译器里直接使用，功能是可以内嵌绘图，并且可以省略掉plt.show()这一步 %matplot…
2021/11/8 14:39:53 人评论 次浏览 -
08 线性回归 + 基础优化算法
线性回归的从零开始实现 包括数据流水线、模型、损失函数和小批量随机梯度下降优化器 # 如果没有d2l包，可以通过下面的语句安装 # !pip install -U d2l# %matplotlib inline可以在Ipython编译器里直接使用，功能是可以内嵌绘图，并且可以省略掉plt.show()这一步 %matplot…
2021/11/8 14:39:53 人评论 次浏览 -
2021-11-06
AssertionError:torch not compiled with cuda enabled错误 原因是因为我安装的torch的是不带cuda的版本，安装包名为Torch-1.0.1-cp37-cp37m-win_amd64.whl的文件，这是Python版本，和系统都和我的匹配，但是这个是不带cuda版本的，可以去官网或者清华大学镜像站点下载带…
2021/11/7 6:14:06 人评论 次浏览 -
2021-11-06
AssertionError:torch not compiled with cuda enabled错误 原因是因为我安装的torch的是不带cuda的版本，安装包名为Torch-1.0.1-cp37-cp37m-win_amd64.whl的文件，这是Python版本，和系统都和我的匹配，但是这个是不带cuda版本的，可以去官网或者清华大学镜像站点下载带…
2021/11/7 6:14:06 人评论 次浏览 -
安装torch+cuda笔记
在安装之前先看一下自己的显卡能不能安装CUDA，如果不能安装或者已经在英伟达官网下载CUDA并安装就跳到第5条。 在torch的官网有一键安装pytorch的命令，官网地址： https://pytorch.org/get-started/locally/ 下面图片中为windows系统，如果有其他操作系统的请自行查看…
2021/11/6 23:41:58 人评论 次浏览 -
安装torch+cuda笔记
在安装之前先看一下自己的显卡能不能安装CUDA，如果不能安装或者已经在英伟达官网下载CUDA并安装就跳到第5条。 在torch的官网有一键安装pytorch的命令，官网地址： https://pytorch.org/get-started/locally/ 下面图片中为windows系统，如果有其他操作系统的请自行查看…
2021/11/6 23:41:58 人评论 次浏览 -
如何生成锚框并在图片可视化
import torch from torch._C import Size from d2l import torch as d2ltorch.set_printoptions(2) #精简打印def multibox_prior(data,sizes,ratios):#生成以每个像素为中心具有不同形状的锚框in_height,int_width = data.shape[-2:]device,num_sizes,num_ratios = data.…
2021/11/5 23:16:04 人评论 次浏览 -
如何生成锚框并在图片可视化
import torch from torch._C import Size from d2l import torch as d2ltorch.set_printoptions(2) #精简打印def multibox_prior(data,sizes,ratios):#生成以每个像素为中心具有不同形状的锚框in_height,int_width = data.shape[-2:]device,num_sizes,num_ratios = data.…
2021/11/5 23:16:04 人评论 次浏览 -
SO-Net中的index_max的功能及具体实现
一、背景在SO-Net的分类模型的Encoder部分，first_pointnet给出特征向量后，有一个求索引的操作：M = node.size()[2]with torch.cuda.device(self.first_pn_out.get_device()):gather_index = index_max.forward_cuda(self.first_pn_out.detach(),min_idx.int(),M).detac…
2021/11/4 23:10:04 人评论 次浏览 -
SO-Net中的index_max的功能及具体实现
一、背景在SO-Net的分类模型的Encoder部分，first_pointnet给出特征向量后，有一个求索引的操作：M = node.size()[2]with torch.cuda.device(self.first_pn_out.get_device()):gather_index = index_max.forward_cuda(self.first_pn_out.detach(),min_idx.int(),M).detac…
2021/11/4 23:10:04 人评论 次浏览
