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

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

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

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

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

 

第一步，通過SSH連接啟動一個管理節(jié)點，并連接進入管理節(jié)點。

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

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

 

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

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      #運行程序

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

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

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

 

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