pytorch
pytorch
sher lock
这个作者很懒,什么都没留下…
展开
-
mask rcnn pytorch 版本跑通笔记
github链接要看代码的,跑别人代码看代码而不是看README教程绝对是事半功倍,看demo.py代码得知下载的模型是放在根目录下的。遇到问题:ImportError: /home/wyy/czh_Project/pytorch-mask-rcnn/nms/_ext/nms/_nms.so: undefined symbol: __cudaRegisterFatBinaryEnd原因是pytorch中的cuda版本与系统安装的CUDA版本号不一致导致的。先用以下命令查询pytorch的cuda版本转载 2021-10-07 16:26:49 · 871 阅读 · 0 评论 -
RuntimeError: module must .... on device cuda:0 (device_ids[0]) but found one of them on device: cpu
链接转载 2021-09-18 13:54:18 · 560 阅读 · 0 评论 -
dcgan-mnist ----pytorch版 代码全注释
github地址main.pyimport argparseimport loggingimport timeimport torchimport torch.nn as nnimport torch.nn.functional as Fimport torch.optim as optimimport numpy as npfrom torch.autograd import Variablefrom torchvision.utils import save_imagefr原创 2021-08-18 20:57:36 · 330 阅读 · 0 评论 -
anaconda安装pytorch环境
1.首先是有anaconda之后在Anaconda Prompt里运行命令2.配置清华镜像不需要看别人花里胡哨的配置方法直接去清华大学开源软件镜像官网配置anaconda镜像,链接如下https://mirror.tuna.tsinghua.edu.cn/help/anaconda/[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-HWbOpyum-1628245639203)(C:\Users\86152\AppData\Roaming\Typora\typora-原创 2021-08-06 18:28:49 · 423 阅读 · 0 评论