pytorch CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`

Bertha 。 2022-11-09 12:53 242阅读 0赞
  1. 环境:
  2. GPU2080ti
  3. ubuntu 18.04
  4. python 3.6
  5. cuda 11.1
  6. cudnn 8.10 for cuda 11.0, 11.1, 11.2
  7. pytorch 对应版本

方法1: 修改batchsize
方法2:改用root用户去运行即可, 原因未知

发表评论

表情:
评论列表 (有 0 条评论,242人围观)

还没有评论,来说两句吧...

相关阅读