遇到的三个问题
(1)下载的pytorch是cpu版本的
解决:配置pytorch源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda install pytorch torchvision cudatoolkit=10.1
(2) 训练到(700/1000)停止,pycharm连接不上服务器的解释器:Couldn’t refresh skeletons for remote interpreter
解决:
删掉pycharm安装文件夹下的skeletons文件夹,
没解决问题的话重新改远程连接的配置,有两个地方
(1)tools -> Deployment -> Configuration connection配置
(2)File -> Setting Project Interpreter
(3) 训练(24/100)停止:RuntimeError: main thread is not in main loop
Tcl_AsyncDelete: async handler deleted by the wrong thread Aborted (core dumped)
解决:
最简单的解决方案是dataloader的num_worker参数设为0,但是加载数据会慢很多
解决方法:utils中,导入matplotlib包的部分进行修改
import matplotlib
#import matplotlib.pyplot as plt
matplotlib.use('Agg') #
from matplotlib import pyplot as plt #