pytorch的安装注意事项

pytorch的官方下载地址是：
https://download.pytorch.org/whl/torch_stable.html
需要安装的是torch、torchvision。版本分为支持cpu和gpu的，而且很多，需要注意寻找。
安装命令是

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的版本是不一样的。这是第一个坑。由于torch包很大，建议先安装torchvision，确认版本后再下载torch。
第二个坑是使用tensorboard会出现错误。网络上给出的方法是使用旧版本的setuptool。这个方法不可取。不如修改tensorboard的init.py文件：

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 而是使用_create_unverified_context。
第四个坑是关于cuda的。估计每个初学者都会碰到。我是在安装tensorflow的时候就遇到了。这次很顺利，没有遇到麻烦。
没想到tensorflow衰落的这么快。我还写了很多东西，现在要从头开始学习pytorch了。程序员是一个苦活，因为有学不完的新东西。