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)
还没有任何评论哟~
