记录一个本人遇到的坑:
os.environ["CUDA-VISIBLE_DEVIECES"]
失效,我一开始想研究一下这条语句的作用,于是在这条语句之前使用了torch.cuda.device_count()
打印了一下设备数,发现os这条语句就不起作用了。
解决:在用os设置cuda之前,不要使用有关cuda
的函数,如果发现失效,那么可能是使用了相关的函数,特别是有的时候import
了一些外部库,你又不知道是怎么实现的,外部库可能就会使用cuda相关的函数,os设置cuda
失败。
os.environ[“CUDA-VISIBLE_DEVIECES“]失效
最新推荐文章于 2024-04-11 11:44:46 发布