pytorch与cuda环境问题

1.运行代码后下面问题

Traceback (most recent call last):
  File "ucf_train_one_shot.py", line 13, in <module>
    from torch.optim.lr_scheduler import StepLR
ModuleNotFoundError: No module named 'torch.optim.lr_scheduler'

使用conda list查看发现pytorch版本为0.1.12，是非常老的版本，需要将pytorch版本升级，其实一开始pytorch安装的版本是0.4.0，cuda是8.0，但是服务器是2080Ti的，需要安装10.0的cuda，结果安装成功之后，将pytorch的版本降为0.1.12（到目前为止，还未想明白是什么原因），这说明pytorch和cuda的安装是有先后的，最好先安装好cuda，再安装pytorch。先使用pytorch官网的安装方法重新安装，未安装成功，卸载pytorch，卸载不掉，出现下面新的问题。

Collecting package metadata (repodata.json): done
Solving environment: |
Found conflicts! Looking for incompatible packages.                                                                                     failed
UnsatisfiableError: The following specifications were found to be incompatible with each other:

还未找到解决方案，欢迎评论。
2.尝试直接在cuda8.0下安装10.0，出现了下面的问题。

Collecting package metadata (repodata.json): done
Solving environment: failed with initial frozen solve. Retrying with flexible solve.

暂时问题还没有找到合适的解决方式，后续会继续观察。目前，在pytorch1.x环境下，轻松卸载8.0安装cuda10.0，所以建议大家尽量使用pytorch1.x环境。