![](https://img-blog.csdnimg.cn/20201014180756738.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
CUDA
生成滞涨网络~
Intel(R) Core(TM) i9-10900X CPU @ 3.70GHz
展开
-
DCN环境配置遇到问题(2080ti以及3080ti)
2080ti的CUDA版本为10.1,3080ti的版本为11.3 需要注意的是30系列显卡适配CUDA版本11以上, 老环境需要进行相应调整。1. 3080ti编译时ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directory查看CUDA版本显示cat /usr/local/cuda/version.tx...原创 2022-01-21 23:27:53 · 2309 阅读 · 0 评论 -
torch.cuda.amp.autocast()使用示例
# 定义模型和优化器model = Net().cuda()optimizer = optim.SGD(model.parameters(), ...)# 在训练最开始定义GradScalar的实例scaler = GradScaler()for epoch in epochs: for input, target in data: optimizer.zero_grad() # 利用with语句,在autocast实例的上下文范围内,进行模型的前向推原创 2021-08-16 15:11:54 · 9212 阅读 · 1 评论