Matebook搭建Pytorch环境,跑UDRL例子

    笔记本上是win10 x64,有两块显卡:uhd 620集成显卡、 mx150 独立显卡,这两块显卡到底用谁由控制面板->NVIDIA控制面板来设置。

    安装Pytorch步骤:

  1. 安装Anaconda3,我选的是python3.7 64-Bit Graphical Installer (466 MB)
  2. cmd 下用pip安装pytorch,由于显卡只支持到CUDA9.2,所以我选stable1.5.1+windows+pip+python+cuda9.2得到:
    pip install torch==1.5.1+cu92 torchvision==0.6.1+cu92 -f https://download.pytorch.org/whl/torch_stable.html
  3. 在 系统环境变量 中加入:
    • )安装路径\ANACONDA
    • )安装路径\ANACONDA\Scripts
    • )安装路径\ANACONDA\Library\bin

    VSCODE环境设置:

  1. 安装pylint插件;
  2. 在settings中增加:
    "python.pythonPath": "D:\\ANACONDA\\python.exe",
    "python.linting.pylintPath":"D:\\ANACONDA\\pkgs\\pylint-2.4.4-py37_0\\Scripts\\pylint",
    // whitelist numpy to remove lint errors
    "python.linting.pylintArgs": [
        "--extension-pkg-whitelist=numpy",
        "--generated-members=numpy.* ,torch.*"
    ],

    UDRL例子:

  1. 从github下载代码(upsidedownRL);
  2. 代码中用到了gym库中的LunarLander,需要安装:
    1. )pip install gym[all]           //安装box2d时出错了
    2. )  pip install box2d              //单独安装
  3.  解压代码,用vscode打开文件夹,选择train.py,运行。

    性能调优:

    性能监视器中发现 mx150 显卡一直0%,CUDA默认调用了集成显卡,所以在nvidia控制面板把 vscode和python单独列为使用高性能GPU,重新打开vscode运行看到GPU1用上了,当然速度依然很慢,截至发稿时还在跑着...

猜你喜欢

转载自blog.csdn.net/xhydongda/article/details/107234301