pytorch

错误定位

3.在终端中打开python环境,分别输入
import torch
from torch.utils.tensorboard import SummaryWriter
summary = SummaryWriter(‘runs/test_simpleClass’)
报错:
W tensorflow/stream_executor/platform/default/dso_loader.cc:60] Could not load dynamic library ‘libcudart.so.11.0’; dlerror: libcudart.so.11.0: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /usr/local/nvidia/lib:/usr/loc
找到问题出现在tensorflow上,tensorflow 的版本要与cudn相匹配(匹配关系可以去官网查),需要卸载版本不匹配的tensorflow版本并安装新版本。
而我的工程不需要tensorflow,因此直接卸载。问题解决。