初步踏入dl领域的小白一枚
之前在训练模型的时候想用tensorboad记录一下训练过程
但是一直在报错,后面顺利解决,想记录一下自己的解决方法
帮助更多的友友不走弯路
这里直接下结论,给出解决方法
俺的电脑配置
cpu:11th Gen Intel(R) Core(TM) i5-11400H @ 2.70GHz
gpu:NVIDA GeForce RTX 3050 Laptop GPU
深度学习环境
python:3.8
pytorch:1.9.0
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
cuda:11.1(自己去官网找把)
cudnn:8005
如果你也和我一样,先按照上面的要求配置好
这里敲重点!!!
如果不能使用tensoboard
就使用tensorboardX,它和tensorboard的使用方式一样
pip install tensorboardX