安装ptan库的时候如果使用pip的方式进行安装,由于ptan库中的setup.py文件中指定的pytorch版本是1.7.0。此时,如果安装的pytorch的版本不符的话,就会被替换为1.7.0的版本,从而导致cuda版本和pytorch的版本不匹配,所以无法使用gpu环境。
检查pytorch gpu环境是否可用
import torch
try:
print(torch.cuda.is_available() and torch.zeros(1).cuda() is not None)
except Exception as e:
print(False)
如果是上述的原因,可以从GitHub上下载ptan的源码进行安装,并删除steup.py文件中对pytorch版本的限定
修改后
requirements = ['torch', 'gym', 'atari-py', 'numpy', 'opencv-python']