出现问题
解决问题
只需要在跑的py文件的前面加上
torch.cuda.current_device()
torch.cuda._initialized = True
就可以正常训练了
反思
找了好多博客,都没有解决这个问题,感觉有时候就会很容易陷入自我怀疑中,逐渐变得暴躁.
但是呢,解决的办法就只有一个一个去尝试,总会解决的
遇到错误也不是我的错,可能是代码或者环境的问题,合理归因吧
更多推荐
RuntimeError:No CUDA GPUs are available
发布评论