
pytorch
文章平均质量分 66
evanna-y
这个作者很懒,什么都没留下…
展开
-
c++编译pytorch layer
envs/py36/lib/python3.6/site-packages/correlation_cuda-0.0.0-py3.6-linux-x86_64.egg/correlation_cuda.cpython-36m-x86_64-linux-gnu.so: undefined symbol: _ZN6caffe26detail37_typeMetaDataInstance_preallocated_32E这个错误查了好久……最后发现,要先import torch 再import co.原创 2020-05-19 23:58:34 · 1518 阅读 · 5 评论 -
安装特定版本的torch-linux
尝试运行https://github.com/jadore801120/attention-is-all-you-need-pytorch时出错,针对torch版本进行了修改。torch官网上最新版本的torch支持的cuda最低版本为92,服务器上cuda版本为9.0,如果安装最新版的torch运行时会报错The NVIDIA driver on your system is too...原创 2019-10-13 20:26:59 · 2377 阅读 · 0 评论 -
Win10 Pytorch Flownet
1. 安装anaconda2. 在anaconda下安装pytorch链接: https://pan.baidu.com/s/1qe4Dxvv-84N_irJRFNg9ow 密码: wsmt进入上面下载文件的上级目录,目录不能有中文,执行下面代码conda install numpy mkl cfficonda install --offline pytorch-0.1.12-py36_0.1....原创 2018-06-06 00:48:02 · 1678 阅读 · 2 评论 -
Pytorch 使用 nii数据做输入数据
使用pix2pix-gan做医学图像合成的时候,如果把nii数据转成png格式会损失很多信息,以为png格式图像的灰度值有256阶,因此直接使用nii的医学图像做输入会更好一点。但是Pythorch中的Dataloader是不能直接读取nii图像的,因此加一个CreateNiiDataset的类。先来了解一下pytorch中读取数据的主要途径——Dataset类。在自己构建数据层时都要基于这个...原创 2019-07-05 18:22:14 · 5172 阅读 · 1 评论