PlatforMAX
平台使用方法
1. 创建交互式开发实例
浏览器访问http://10.126.62.119/
输入用户名密码登录,进入主页
点击菜单: AI任务→交互式开发→创建
执行器选择Terminal
,
设置密码
,
加速器
选择NVIDIA_GeForce_RTX_4090
镜像
选择集群→tensorflow-gpu
提交
2. 远程连接服务器
回退到AI任务
→交互式开发
点击刚创建好的Terminal实例中的复制ssh命令
Win+R
打开cmd
终端
粘贴ssh命令
并执行
输入前边设定的密码
,连接成功
3. 配置Python环境
更新apt-get:apt-get update
-
若出现有关
publickey
类似错误:The following signatures couldn't be verified because the public key is not available: NO_PUBKEY A4B469963BF863CC
则运行:
apt-key adv --keyserver keyserver.ubuntu.com --recv-keys A4B469963BF863CC
查看python
版本:
python3 --version
安装对应python
版本的venv
模块:
apt-get install python3.8-venv
创建虚拟环境,并命名为pytorch_env
:
python3 -m venv pytorch_env
激活虚拟环境:
source pytorch_env/bin/activate
查看cuda
版本:
nvidia-smi
下载对应版本的torch
包:
wget https://download.pytorch.org/whl/cu124/torch-2.4.0%2Bcu124-cp38-cp38-linux_x86_64.whl#sha256=2d31274281660d6a855ee7891cfdab0c52bdf0e920fadeba3c661da6fda228fc
安装torch
包:
pip install ./torch-2.4.0+cu124-cp38-cp38-linux_x86_64.whl
安装numpy
包:
pip install numpy
查看安装情况:
pip list
测试:
进入python
控制台:
python3
导入torch
包:
import torch
检查cuda
是否可用:
torch.cuda.is_available()