
深度学习
光子乘羽
我从来没有长大,也从来没有停止过成长。
展开
-
failed to allocate 2.00G (2147483648 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY: out of memory
显存充足,但是却出现CUDA error:out of memory错误添加:import os# 设置程序使用的GPU编号:我用6号和7号显卡os.environ['CUDA_VISIBLE_DEVICES'] = '0,1'import numpy as npimport kerasimport keras.backend as Kimport tensorflow.compat.v1 as tftf.disable_v2_behavior()import keras.b原创 2022-03-01 20:59:49 · 4888 阅读 · 2 评论 -
Could not load library cudnn_cnn_infer64_8.dll. Error code 126
Could not load library cudnn_cnn_infer64_8.dll. Error code 126Please make sure cudnn_cnn_infer64_8.dll is in your library path!C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.5\lib下有对应文件,但是依旧报错。解决:cudnn版本从8.3改为8.2Index of /anaconda/pkgs/main/原创 2022-01-22 19:55:51 · 1794 阅读 · 0 评论 -
PyG (PyTorch Geometric) GNN 笔记
https://pytorch-geometric.readthedocs.io/en/latest/index.htmlINTRODUCTION BY EXAMPLEhttps://pytorch-geometric.readthedocs.io/en/latest/notes/introduction.htmlA graph is used to model pairwise relations (edges) between objects (nodes). A single graph in原创 2021-12-15 20:26:45 · 3647 阅读 · 0 评论 -
ModuleNotFoundError: No module named ‘torch_sparse‘
https://github.com/rusty1s/pytorch_sparseYou can now install pytorch-sparse via Anaconda for all major OS/PyTorch/CUDA combinations ???? Given that you have pytorch >= 1.8.0 installed, simply runconda install pytorch-sparse -c pyg原创 2021-12-02 17:34:20 · 1757 阅读 · 0 评论 -
tensorflow版本问题 AttributeError: module ‘tensorflow‘ has no attribute ‘global_variables_initializer‘
import tensorflow as tf改为:import tensorflow.compat.v1 as tftf.disable_v2_behavior()原创 2021-09-27 10:51:29 · 427 阅读 · 0 评论 -
docopt.DocoptLanguageError: “usage:“ (case-insensitive) not found.
arguments = docopt(doc) 报错:docopt.DocoptLanguageError: “usage:” (case-insensitive) not found.在“”"“”"直接添加Usage:,Options:等说明。例:“”"Using multilayer perceptron (MLP) and ensemble learning to classify the ASD and TC based on single brain atlas.Usage:n原创 2021-09-27 10:29:28 · 408 阅读 · 0 评论 -
ml-agents在导出的unity环境部署训练模型onnx
import onnximport mlagentsimport onnxruntime as ortimport numpy as npimport timeimport torchimport mathimport gymimport tensorflow as tffrom gym_unity.envs import UnityToGymWrapperfrom mlagents_envs.environment import UnityEnvironmentfrom mlagen原创 2021-09-14 11:45:23 · 969 阅读 · 0 评论 -
onnx安装与使用
官网:https://github.com/onnx/onnxpip install onnx读取.onnx文件:import onnxmodel = onnx.load(r’C:\Users\Win10\Desktop\Pyramids.onnx’ )print(model)原创 2021-07-05 11:05:11 · 28505 阅读 · 0 评论