阅读量:2
在Ubuntu上安装PyTorch,可通过以下两种常见方式下载安装包:
- 使用pip安装:访问PyTorch官方网站,根据系统、CUDA版本等信息生成安装命令,如在有CUDA 11.8支持的情况下,可执行命令
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118来下载安装。若无GPU,可使用pip3 install torch torchvision torchaudio命令安装CPU版本。 - 使用conda安装:先安装Miniconda或Anaconda,然后创建虚拟环境并激活。以安装CUDA 11.8版本的PyTorch为例,可执行命令
conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch -c nvidia,无GPU则使用conda install pytorch torchvision torchaudio cpuonly -c pytorch命令。
以上就是关于“Ubuntu PyTorch安装包在哪里下载”的相关介绍,筋斗云是国内较早的云主机应用的服务商,拥有10余年行业经验,提供丰富的云服务器、租用服务器等相关产品服务。云服务器资源弹性伸缩,主机vCPU、内存性能强悍、超高I/O速度、故障秒级恢复;电子化备案,提交快速,专业团队7×24小时服务支持!
简单好用、高性价比云服务器租用链接:https://www.jindouyun.cn/product/cvm