PyTorch¶

简介¶

PyTorch 是一个 Python 优先的深度学习框架，也是使用 GPU 和 CPU 优化的深度学习张量库，能够在强大的 GPU 加速基础上实现张量和动态神经网络。

安装环境¶

使用方法¶

#加载软件环境
module load anaconda/3-2023.09
source activate pytorch
#运行python
(pytorch) [@login01 ~]$ python
Python 3.11.5 (main, Sep 11 2023, 13:54:46) [GCC 11.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.get_num_threads()
64
>>>

作业提交¶

hpckapok1 pytorch cpu版本使用方法¶

step1、创建slurm脚本pytorch.sh

#!/bin/bash
#SBATCH -J test
#SBATCH -p cpuXeon6458
#SBATCH -o %j.out
#SBATCH -e %j.err
#SBATCH -N 1
#SBATCH --cpus-per-task=64
module load anaconda/3-2023.09
source activate pytorch
python -c "import torch; print(torch.__version__)"

step2、提交作业

sbatch  pytorch.sh

hpckapok1 pytorch gpu版本使用方法¶

step1、创建slurm脚本pytorch.sh

#!/bin/bash
#SBATCH -J test
#SBATCH -p gpuA800
#SBATCH -o %j.out
#SBATCH -e %j.err
#SBATCH -N 1
#SBATCH -n 12
#SBATCH --gres=gpu:1
module load anaconda/3-2023.09
source activate pytorch_cuda11.8
python -c "import torch; print(torch.__version__); print(torch.zeros(10, 10).device)"

step2、提交作业

sbatch  pytorch.sh

参考资料¶

Contributor：qbxie

最后更新: 2025 年 06 月 30 日