pytorch

PyTorch是一个开源的Python机器学习库,基于Torch,用于自然语言处理等应用程序。

1. 安装

Step 1. 导入conda环境;

在自己的HOME目录安装conda或者通过module导入conda环境;

Step 2.创建虚拟环境并安装pytorch,这里安装1.6.0版本;

conda create -n pytorch-1.6.0 pytorch=1.6.0 torchvision cudatoolkit=10.1 -c pytorch

Step 3. 进入和退出创建好的虚拟环境;

source activate pytorch-1.6.0
conda deactivate

2. 提交作业

Step 1. 创建工作目录并进入;

mkdir pytorchJob1
cd pytorchJob1

Step 2. 将运行pytorch需要的相关文件上传到该文件夹下,这里创建一个简单的test.py程序,用于检测GPU是否可用;

import torch
print(torch.cuda.is_available())
torch.zeros(1).cuda()

Step 3. 在该文件夹下编写作业脚本,命名为pytorchJob1.sh,脚本内容如下;

#!/bin/bash
#SBATCH -o job.%j.out
#SBATCH --partition=GPU
#SBATCH -J pytorch_job_1
#SBATCH -N 1
#SBATCH --ntasks-per-node=2
#SBATCH --gres=gpu:1
#SBATCH --qos=low

source activate pytorch-1.6.0
python test.py

Step 4. 提交作业;

sbatch pytorchJob1.sh

results matching ""

    No results matching ""