摘要: 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 阅读(309) 评论(0) 推荐(0) 编辑
摘要: 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 阅读(58) 评论(0) 推荐(0) 编辑
摘要: Convolution Layers torch.nn.Conv2d 阅读全文
posted @ 2022-10-03 19:29 watsono 阅读(21) 评论(0) 推荐(0) 编辑
摘要: tensorboard tensorboard --logdir=目录 --port=端口 --samples_per_plugin images=99999999 阅读全文
posted @ 2022-10-02 16:46 watsono 阅读(489) 评论(0) 推荐(0) 编辑
摘要: Jupyter Anaconda虚拟环境中Jupyter的安装 conda install nb_jupyter 进入Jupyter环境 jupyter notebook 阅读全文
posted @ 2022-09-26 21:54 watsono 阅读(15) 评论(0) 推荐(0) 编辑
摘要: test1.py import tensorflow as tf import numpy as np def add_layer(inputs, in_size, out_size, activation_function=None): # inputs.shape=[None,1],in_siz 阅读全文
posted @ 2022-09-25 20:09 watsono 阅读(30) 评论(1) 推荐(0) 编辑
摘要: #1 DCU #!/bin/bash #SBATCH --job-name='test' #SBATCH --partition=hebhdnormal #SBATCH --nodes=1 #SBATCH --ntasks-per-node=16 #SBATCH --gres=dcu:2 #使用两块 阅读全文
posted @ 2022-09-25 20:05 watsono 阅读(223) 评论(0) 推荐(0) 编辑
摘要: Anaconda基础命令 查看自己配置的环境 conda env list 配置一个新的环境 conda create -n 环境的名字 python=版本号 进入和退出环境 conda activate 环境名 #激活环境 conda deactivate #退出环境 删除环境 conda rem 阅读全文
posted @ 2021-03-05 16:08 watsono 阅读(81) 评论(0) 推荐(0) 编辑
摘要: 曙光平台 提交作业程序(***.sh) #!/bin/bash #SBATCH --job-name='test' #任务名 #SBATCH --partition=lzhgnormal #可访问队列名称 #SBATCH --nodes=1 #使用节点个数 #SBATCH --ntasks-per- 阅读全文
posted @ 2020-04-30 10:03 watsono 阅读(86) 评论(2) 推荐(0) 编辑