
deep-learning
周迪新
好记性不如烂笔头
展开
-
OSError: (External) CUDA error(222), the provided PTX was compiled with an unsupported toolchain..
安装了CUDA 10.1, 11.1, 11.3, 11.4, 11.8, 12.0都不行。检查一下已经安装的NVIDIA驱动,返回为空的话,说明没有驱动,直接跳过这一步。本来的驱动 NVIDIA 470。原创 2023-09-22 15:04:40 · 3501 阅读 · 4 评论 -
PyTorch Error: Expected more than 1 value per channel when training
python import parent directory原创 2022-08-24 17:02:21 · 418 阅读 · 0 评论 -
Ubuntu 20.04 安装CUDA11.1 和cudnn 8.0.5
Install CUDAcuda 11.1百度网盘链接密码: bo6fsudo sh cuda_10.0.130_410.48_linux.run安装成功后跳出============ Summary ============Driver: Not SelectedToolkit: Installed in /usr/local/cuda-11.1/Samples: Not SelectedPlease make sure that - PATH includes原创 2022-04-12 12:54:29 · 3639 阅读 · 0 评论 -
tensorflow-gpu RDKit dockerfile
FROM nvidia/cuda:11.1.1-base-ubuntu18.04 LABEL maintainer "fanxp <897488736@qq.com>" WORKDIR /root # CUDA installRUN apt-get update && apt-get install -y --no-install-recommends cuda-minimal-build-11-1 \ libcusparse-dev-11-1 libcub原创 2021-10-11 14:11:13 · 286 阅读 · 0 评论 -
pytorch 保存模型出错, pickler.dump(obj) Error
那个 把 报错中的 serialization.py 文件开头的 PROTOCOL=2 改成 PROTOCOL=4我的报错中 serialization.py 的 路径是lib/python3.8/site-packages/torch/serialization.py原创 2021-10-05 18:48:21 · 475 阅读 · 1 评论 -
WARNING:tensorflow:Entity <bound method GRUCell.call of <tensorflow.python.ops.rnn_cell_impl.GRUCell
我是在 Anaconda 环境下的。运行conda install -c anaconda gast=0.2.2原创 2021-09-22 15:21:01 · 466 阅读 · 0 评论 -
使用 ONNX 模型做预测
import ioimport numpy as npfrom torch import nnimport torch.utils.model_zoo as model_zooimport torch.onnx定义模型import torch.nn as nnimport torch.nn.init as initclass SuperResolutionNet(nn.Module): def __init__(self, upscale_factor, inplace=Fa原创 2021-09-13 20:07:58 · 2893 阅读 · 0 评论 -
查看 ONNX 模型 weights
model_path = 'model.onnx'model = onnx.load(model_path) weights = [] names = []for t in model.graph.initializer: weights.append(numpy_helper.to_array(t)) names.append(t.name)model_weights = {}for name, weight in zip(names, weights):原创 2021-09-13 16:56:35 · 2726 阅读 · 0 评论 -
一行命令配置深度学所需所有环境PyTorch, TensorFlow, CUDA, cuDNN, and NVIDIA Drivers.
参考https://lambdalabs.com/lambda-stack-deep-learning-softwareGitHub运行下面命令在新安装的Ubuntu(20.04, 18.04, or 16.04)LAMBDA_REPO=$(mktemp) && \wget -O${LAMBDA_REPO} https://lambdalabs.com/static/misc/lambda-stack-repo.deb && \sudo dpkg -i ${LAM原创 2021-08-06 10:39:21 · 127 阅读 · 0 评论 -
PyTorch 靠谱的模型可视化教程
Netron安装pip install netron使用~$ python3>>> import netron>>> netron.start('/y/home/zdx/model.pth')Serving '/y/home/zdx/model.pth' at http://localhost:8080('localhost', 8080)>>> 打开浏览器,把上面的 http://localhost:8080,用浏览器打开左上原创 2021-07-29 19:27:17 · 1869 阅读 · 0 评论 -
Default process group has not been initialized, please make sure to call init_process_group
在使用 PyTorch Lightning 在导入模型时出现该错误的时候如果你在训练时使用 了ddp加速。去掉ddp加速即可。原创 2021-07-20 09:27:29 · 1906 阅读 · 0 评论 -
在Ubuntu 16.04上安装pytorch gpu和tensorflow gpu版[Anaconda]
Install NVIDIAsudo vim /etc/modprobe.d/blacklist.confadd the following lines to the end of the file blacklist.confblacklist vga16fbblacklist nouveaublacklist rivafbblacklist rivatvblacklist nv...原创 2019-10-17 11:08:21 · 1079 阅读 · 0 评论 -
配置一台用于深度学习的个人工作站系统的实验记录[主机可以买现成的或自己配,裸机即可]
一、安装ubuntu18.041.1 使用UltraISO制作ubuntu启动盘,ultraiso下载入口Tips:u盘应大于4G, 且需要格式化为FAT32最后,直接点击“写入”,等待完成即可。1.2 开始安装在机器上安装ubuntu1)选择U盘启动方式(msi开机按F11),进入ubuntu安装界面2)选择最小安装,为图形或无线硬件选项不选3)安装类型选择:清除整个磁盘...原创 2021-01-14 09:33:33 · 2465 阅读 · 0 评论 -
用于Ubuntu 16.04和18.04的TensorFlow,Keras,Caffe,Caffe,CUDA,cuDNN和NVIDIA驱动程序的单行安装
打开终端:分别输入LAMBDA_REPO=$(mktemp) && \wget -O${LAMBDA_REPO} https://lambdalabs.com/static/misc/lambda-stack-repo.deb && \sudo dpkg -i ${LAMBDA_REPO} && rm -f ${LAMBDA_REPO} &am...原创 2019-03-23 23:24:17 · 280 阅读 · 0 评论 -
如何在Ubuntu-16.04 / 18.04上为 RTX 2080 Ti GPU 安装Nvidia驱动和cuda-10.0
https://medium.com/@avinchintha/how-to-install-nvidia-drivers-and-cuda-10-0-for-rtx-2080-ti-gpu-on-ubuntu-16-04-18-04-ce32e4edf1c0END原创 2019-03-23 21:23:02 · 2129 阅读 · 1 评论