Advertisement

pytorch的安装注意事项

阅读量:

访问PyTorch的官方下载页面:https://download.pytorch.org/whl/torch_stable.html

复制代码
    pip install torch-1.10.0+cu113-cp39-cp39-win_amd64.whl
    pip install torchvision-0.12.0+cpu-cp39-cp39-win_amd64.whl

需要特别注意的是 torch 和 torchvision 的版本分别为不同的软件包。

网络上给出的方法是使用旧版本的 setup 工具

复制代码
    import tensorboard
    #from setuptools import distutils
    from distutils import version as vvv
    
    #LooseVersion = distutils.version.LooseVersion
    LooseVersion = vvv.LooseVersion
    if not hasattr(tensorboard, '__version__') or LooseVersion(tensorboard.__version__) < LooseVersion('1.15'):
    raise ImportError('TensorBoard logging requires TensorBoard version 1.15 or above')
    
    #del distutils
    del vvv
    del LooseVersion
    del tensorboard
    
    from .writer import FileWriter, SummaryWriter  # noqa: F401
    from tensorboard.summary.writer.record_writer import RecordWriter  # noqa: F401

通过直接调用distutils模块来实现功能。另一个常见问题涉及 SSL 链接的问题。由于证书验证的问题可能导致 SSL 连接失败。解决方法是在 SSL.py 文件中进行修改。

复制代码
    #_create_default_https_context = create_default_context
    _create_default_https_context = _create_unverified_context

在805行处不再依赖create_default_context而是创建未验证上下文

全部评论 (0)

还没有任何评论哟~