远程服务器(已配置)运行深度学习算法踩坑
查看可使用的显卡nvidia-smi由上图可知显卡0目前空闲,可以用来运行我们的代码根据代码文档激活环境我这里使用anaconda激活了预先安装的pytorch1.6以及python3.6的环境,操作如下:conda activate torch16py36选择显卡运行代码使用单个GPU加速计算若代码中有设定默认显卡,则设置默认显卡为0,直接开始运行python xxxx.py若没有设定默认显卡,可以在命令前设置空闲显卡可见,屏蔽其他显卡CUDA_VISIBLE_DEVIC
原创
2020-10-12 11:10:44 ·
1378 阅读 ·
0 评论