摘要:
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. 阅读全文
摘要:
GPU训练方式1 网络模型 数据(输入,标注) 损失函数 加上.cuda() GPU训练方式2 # 测试是否有gpu device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') # 如果有多张显卡,可以指定显卡 # de 阅读全文
摘要:
Convolution Layers torch.nn.Conv2d 阅读全文
摘要:
tensorboard tensorboard --logdir=目录 --port=端口 --samples_per_plugin images=99999999 阅读全文
摘要:
Jupyter Anaconda虚拟环境中Jupyter的安装 conda install nb_jupyter 进入Jupyter环境 jupyter notebook 阅读全文
摘要:
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 阅读全文
摘要:
#1 DCU #!/bin/bash #SBATCH --job-name='test' #SBATCH --partition=hebhdnormal #SBATCH --nodes=1 #SBATCH --ntasks-per-node=16 #SBATCH --gres=dcu:2 #使用两块 阅读全文
摘要:
Anaconda基础命令 查看自己配置的环境 conda env list 配置一个新的环境 conda create -n 环境的名字 python=版本号 进入和退出环境 conda activate 环境名 #激活环境 conda deactivate #退出环境 删除环境 conda rem 阅读全文
摘要:
曙光平台 提交作业程序(***.sh) #!/bin/bash #SBATCH --job-name='test' #任务名 #SBATCH --partition=lzhgnormal #可访问队列名称 #SBATCH --nodes=1 #使用节点个数 #SBATCH --ntasks-per- 阅读全文