1. import torch
torch.cuda.is_available() 返回False, 发现需要使用torch对应的GPU版本:
https://download.pytorch.org/whl/torch_stable.html
查看版本:
import torch
torch.__version__
import torchvision
torchvision.__version__
注意下面红色的CUDA 11.1版本和python版本3.7。
cu111/torch-1.8.1%2Bcu111-cp37-cp37m-win_amd64.whl
torchvision-0.9.1+cu111-cp37-cp37m-win_amd64.whl
2. 出现了如下错误:
Error loading “\site-packages\torch\lib\caffe2_detectron_ops_gpu.dll“ or one of its dependencies.
下载VC_redist.x64.exe,安装相关的dll文件即可。
pip install intel-openmp
另外出现页面分配不足时,需要调整硬盘对应的虚存:
3. 训练时出现下面错误, 学要修改nw=0:
ForkingPickler(file, protocol).dump(obj)
BrokenPipeError: [Errno 32] Broken pipe
create_dataloader
nw = min([os.cpu_count(), batch_size if batch_size > 1 else 0, workers]) # number of workers
nw = 0