无法从Torchvision在我公司的GPU群集上加载模型

时间:2020-06-28 09:28:50

标签: docker pytorch gpu cluster-computing torchvision

代码:

from torchvision.models import vgg16
vgg_model = vgg16(pretrained=True).features[:16]

日志输出:

Traceback (most recent call last):
  File "train.py", line 95, in <module>
    vgg_model = vgg16(pretrained=True).features[:16]
  File "/opt/conda/lib/python3.7/site-packages/torchvision/models/vgg.py", line 150, in vgg16
    return _vgg('vgg16', 'D', False, pretrained, progress, **kwargs)
  File "/opt/conda/lib/python3.7/site-packages/torchvision/models/vgg.py", line 93, in _vgg
    progress=progress)
  File "/opt/conda/lib/python3.7/site-packages/torch/hub.py", line 480, in load_state_dict_from_url
    os.makedirs(model_dir)
  File "/opt/conda/lib/python3.7/os.py", line 213, in makedirs
    makedirs(head, exist_ok=exist_ok)
  File "/opt/conda/lib/python3.7/os.py", line 213, in makedirs
    makedirs(head, exist_ok=exist_ok)
  File "/opt/conda/lib/python3.7/os.py", line 223, in makedirs
    mkdir(name, mode)
PermissionError: [Errno 13] Permission denied: '/.cache'

我必须在GPU群集上运行我的代码,并且我无权在GPU群集上修改这些docker映像。我该怎么办?

0 个答案:

没有答案