笔记本上是win10 x64,有两块显卡:uhd 620集成显卡、 mx150 独立显卡,这两块显卡到底用谁由控制面板->NVIDIA控制面板来设置。
安装Pytorch步骤:
- 安装Anaconda3,我选的是python3.7 64-Bit Graphical Installer (466 MB);
- 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
- 在 系统环境变量 中加入:
- )安装路径\ANACONDA
- )安装路径\ANACONDA\Scripts
- )安装路径\ANACONDA\Library\bin
VSCODE环境设置:
- 安装pylint插件;
- 在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例子:
- 从github下载代码(upsidedownRL);
- 代码中用到了gym库中的LunarLander,需要安装:
- )pip install gym[all] //安装box2d时出错了
- ) pip install box2d //单独安装
- 解压代码,用vscode打开文件夹,选择train.py,运行。
性能调优:
性能监视器中发现 mx150 显卡一直0%,CUDA默认调用了集成显卡,所以在nvidia控制面板把 vscode和python单独列为使用高性能GPU,重新打开vscode运行看到GPU1用上了,当然速度依然很慢,截至发稿时还在跑着...