深度学习中的基本命令知识Linux以及pytorch

nvidia-smi----查看GPU占用
import os
os.environ[“CUDA_VISIBLE_DEVICES”] = ‘1’
----改变训练GPU位置块号
nvcc -v----查看cuda版本
torch.max(0)和 torch.max(1)----分别找出tensor里每列/每行中最大的值并返回索引

你可能感兴趣的:(人工智能,深度学习,python)