PyTorch 是一個(gè) Python 優(yōu)先的深度學(xué)習(xí)框架，也是使用 GPU 和 CPU 優(yōu)化的深度學(xué)習(xí)張量庫，能夠在強(qiáng)大的 GPU 加速基礎(chǔ)上實(shí)現(xiàn)張量和動態(tài)神經(jīng)網(wǎng)絡(luò)。

在使用PyTorch的時(shí)候經(jīng)常會遇到一些問題，比如虛擬環(huán)境配置錯(cuò)誤（沒有安裝TensorBoard），報(bào)錯(cuò)搞不懂問題出在哪等

或者是一些硬件上的問題，自己的電腦跑不動之類的。

這種時(shí)候就發(fā)現(xiàn)了北鯤云超算平臺（不是

想要在北鯤云上使用Pytorch還是很簡單的，就像標(biāo)題說的，不用自己配置環(huán)境，可以直接使用。

第一步，通過SSH連接啟動一個(gè)管理節(jié)點(diǎn)，并連接進(jìn)入管理節(jié)點(diǎn)。

第二步，提供文件傳輸上傳輸入文件，這里我們使用的是Linux數(shù)據(jù)傳輸

在文件傳輸頁面，點(diǎn)擊“新建”上傳文件或文件夾，您也可以進(jìn)入到自定義目錄，點(diǎn)擊“新建”上傳到自定義目錄

第三步，按照第一步中的連接管理節(jié)點(diǎn)后創(chuàng)建作業(yè)目錄并進(jìn)入

?

mkdir?pytorchJob1
cd?pytorchJob1

?

第四步，在之前創(chuàng)建的文件夾下執(zhí)行腳本

?

#!/bin/bashmodule?add?Anaconda3/2020.02??#加載Anaconda3軟件
source?activate?pytorch-1.5???#激活pytorch環(huán)境
python?mnist.py?>?py.log??????#運(yùn)行程序

?

第五步 使用sbatch命令提交作業(yè)，這里要提示的是，北鯤云使用的是slurm作業(yè)管理系統(tǒng)

?

sbatch?-p?g-t4-1?-c?4?pytorch.sh

?

此處為提交任務(wù)到一張T4卡的GPU節(jié)點(diǎn)運(yùn)行。

這只是舉例配置，平臺還有更多配置可以供大家使用。還有更多信息和案例可以歡迎來我們官網(wǎng)看看

審核編輯 黃昊宇