报错:
在import albumentations as A的时候,出现:anaconda服务似乎挂掉了,但是会立刻重启的。如下图
原因:可能albumentations这个库跟显存有关。
解决办法:在加上GPU的指定后就解决了问题。在代码中加一个使用GPU的指定,如:
os.environ['CUDA_VISIBLE_DEVICES'] = '0'
搞定。
欢迎关注公众号:
报错:
在import albumentations as A的时候,出现:anaconda服务似乎挂掉了,但是会立刻重启的。如下图
原因:可能albumentations这个库跟显存有关。
解决办法:在加上GPU的指定后就解决了问题。在代码中加一个使用GPU的指定,如:
os.environ['CUDA_VISIBLE_DEVICES'] = '0'
搞定。
欢迎关注公众号: