远程服务器（已配置）运行深度学习算法踩坑

查看环境

查看cuda是否可用

在python中使用torch.cuda.is_available(),返回True，说明cuda可以正常使用

conda查看环境

控制台里使用conda info -e可以查看conda已经配置过的环境

查看可使用的显卡

nvidia-smi

由上图可知显卡0目前空闲，可以用来运行我们的代码

根据代码文档激活环境

我这里使用anaconda激活了预先安装的pytorch1.6以及python3.6的环境，操作如下：

conda activate torch16py36

指定显卡运行代码

使用单个GPU加速计算

若代码中有设定默认显卡，则设置默认显卡为0,直接开始运行

python xxxx