10 2022 档案

torch==1.9.1+rocm4.0.1
摘要：AMD显卡安装pytorch pytorch v1.8.0 # ROCM 4.0.1 (Linux only) pip install torch==1.8.0+rocm4.0.1 -f https://download.pytorch.org/whl/rocm4.0.1/torch_stable. 阅读全文
posted @ 2022-10-31 21:25 watsono 阅读(345) 评论(0) 推荐(0) 编辑
利用GPU进行模型训练
摘要：GPU训练方式1 网络模型 数据（输入，标注） 损失函数 加上.cuda() GPU训练方式2 # 测试是否有gpu device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') # 如果有多张显卡，可以指定显卡 # de 阅读全文
posted @ 2022-10-07 14:02 watsono 阅读(71) 评论(0) 推荐(0) 编辑
卷积层的参数计算
摘要：Convolution Layers torch.nn.Conv2d 阅读全文
posted @ 2022-10-03 19:29 watsono 阅读(24) 评论(0) 推荐(0) 编辑
tensorboard的step显示不全
摘要：tensorboard tensorboard --logdir=目录 --port=端口 --samples_per_plugin images=99999999 阅读全文
posted @ 2022-10-02 16:46 watsono 阅读(656) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示