PyTorch 是一個 Python 優先的深度學習框架，也是使用 GPU 和 CPU 優化的深度學習張量庫，能夠在強大的 GPU 加速基礎上實現張量和動態神經網絡。

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

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

這種時候就發現了北鯤云超算平臺（不是

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

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

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

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

第三步，按照第一步中的連接管理節點后創建作業目錄并進入

?

mkdir?pytorchJob1
cd?pytorchJob1

?

第四步，在之前創建的文件夾下執行腳本

?

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

?

第五步 使用sbatch命令提交作業，這里要提示的是，北鯤云使用的是slurm作業管理系統

?

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

?

此處為提交任務到一張T4卡的GPU節點運行。

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

審核編輯 黃昊宇