安装ptan库后不能使用pytorch gpu

安装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']

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值