画图【无标题】

画模型图 ：
https://zhuanlan.zhihu.com/p/531619117

画卷积

画resnet
自己画一些简单的框框图

raise ValueError(“Unknown CUDA arch ({}) or GPU not supported”.format(arch))
换了个更高版本的torch

pip install torch1.8.1+cu101 torchvision0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
1.6.0到1.8

export CUDA_HOME=‘/data1/xyj/i2t/AdvancedLiterateMachinery/DocumentUnderstanding/LORE-TSR/’
export LD_LIBRARY_PATH=$CUD{A}_{H}OME/lib64:$LD_LIBRARY_PATH

export CUDA_HOME=/usr/local/cuda-10.1
LD_LIBRARY_PATH该设置为什么
sh make.sh

./make.sh

export CUDA_HOME=/usr/local/cuda-11.4
export LD_LIBRARY_PATH=$CUD{A}_{H}OME/lib64:$LD_LIBRARY_PATH
./make.sh

export CUDA_HOME=/usr/local/cuda-10.1
export PATH=$CUD{A}_{H}OME/bin:$PATH
export LD_LIBRARY_PATH=$CUD{A}_{H}OME/lib64:$LD_LIBRARY_PATH

import torch
print(torch.version)
print(torch.cuda.is_available()) # cuda是否可用
print(torch.version.cuda) # cuda版本