pytorch查看loss曲线_Pytorch初体验

安装

安装Pytorch可以根据官网上的指令进行，但是现在稳定版本是1.5的默认没有CUDA10.0的版本，只有选择安装之前的版本，点击Get Start下面的Previous versions of PyTorch进去。由于我电脑上切换了清华源，选择Wheel的形式安装，找到安装torch==1.2+cuda10.0的指令，在清华源上下载几分钟就安装好了，也可选择其他CUDA版本以及torch版本，看电脑环境实际选择。

pytorch 安装

#CUDA 10.0
pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html

Optional: 查看是否安装好时候，发现出现了操作系统无法运行 %1的错误。

>>> import torch
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "D:ProgramDataAnaconda3libsite-packagestorch__init__.py", line 81, in <module>
    from torch._C import *
ImportError: DLL load failed: 操作系统无法运行 %1。

根据git上pytorch的ISSUE31324是由于系统加载了System32目录下的libiomp5md.dll，而非安装Pytorch或者Python的目录，将C:WindowsSystem32目录下的libiomp5md.dll删除(建议备份好)就可以了。