NVIDIA DGX Spark | Ubuntu cuda13.0安装Pytorch GPU版本

PS: 费了半天,最终安装的是早期版本:可直接看第6点进行尝试,节省时间!

  1. NVIDIA DGX Spark内置了CUDA 13.0。目前Pytorch官方提供最新,针对CUDA 13.0的Pytorch选择如下(安装会失败):
  1. 输入上述命令安装失败,显示如下问题:

Could not fetch URL https://download.pytorch.org/whl/cu130/torch/: There was a problem confirming the ssl certificate: HTTPSConnectionPool(host='download.pytorch.org', port=443): Max retries exceeded with url: /whl/cu130/torch/ (Caused by SSLError(SSLEOFError(8, 'SSL: UNEXPECTED_EOF_WHILE_READING EOF occurred in violation of protocol (_ssl.c:1017)'))) - skipping

具体如下图所示:

原因:DS查询,说是如下原因。。。

  1. 正确安装:首先通过Pytorch的 Previous PyTorch Versions查询CUDA 13.0版本的Pytorch。然后通过anaconda官方 网站查询相关资源,如下图,查得torch=2.9.1版本:

4.根据上述操作,查得安装命令如下:

复制代码
conda install conda-forge::pytorch-gpu
  1. 安装后测试如下,成功!
  1. 安装成功后通过conda list查询,选择pytorch-gpu 2.9.1版本,实际安装的是2.6.0版本,如下图:
  1. 那这样是否可以直接用Pytorch官网中的2.6.0版本进行安装呢?(PS. 相应cuda版本为12.6),具体步骤如下图可见:(所以在NVIDIA DGX Spark的cuda为13.0的版本上,安装Pytorch官方的2.6.0版本是否就可以了呢?如果安装可以使用,麻烦留个言哈!
相关推荐
晨枫阳29 分钟前
@changesets/cli是什么?哪些情况下需要使用?怎么使用
linux·运维·ubuntu
cc复CC1 小时前
从零开始手写 Spark 07|划分 Stage 与 DAG
spark
%472 小时前
DAY 38
人工智能·pytorch·深度学习
hkNaruto6 小时前
【Linux】Ubuntu 22.04 LTS GNOME 原生远程桌面配置:告别vnc、nomachine
linux·运维·ubuntu
大菠萝爱上小西瓜8 小时前
【大数据实战】Spark 2.2.2 完全分布式集群搭建
大数据·分布式·spark
Είναι η κοπέλα8 小时前
PyTorch 模型导出与部署实战:ONNX + onnxruntime(可直接落地)
人工智能·pytorch·python
Eloudy9 小时前
Ubuntu 22.04 从源码编译安装 ROS 2 Jazzy 笔记
linux·笔记·ubuntu
呆萌很9 小时前
PyTorch CosineAnnealingLR的T_max和eta_min参数设置
人工智能·pytorch·python
盼小辉丶10 小时前
PyTorch强化学习实战——用预训练语言模型和ChatGPT玩转文本游戏
pytorch·深度学习·语言模型·chatgpt·强化学习
Lee_jerome19 小时前
从 PyTorch 权重到 RK3588 板端推理:ResNet18 二分类模型完整部署教程
pytorch·边缘计算·rk3588·模型部署·onnx·resnet18·int8量化