本文主要是介绍解决方案:在autodl环境下安装torch被killed掉,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
文章目录
- 一、现象
- 二、解决方案
一、现象
平台:autodl
镜像:PyTorch 2.0.0 Python 3.8(ubuntu20.04) Cuda 11.8
GPU:A40(48GB) * 1
CPU:15 vCPU AMD EPYC 7543 32-Core Processor
内存:80GB
安装torch:1.13.0环境,发现执行下面的命令行都不行**(不行的话,换个时间点再跑跑试试,或换个镜像)**
pip install torch==1.13.0+cu117 torchvision==0.14.0+cu117 torchaudio==0.13.0 --extra-index-url https://download.pytorch.org/whl/cu117 -i http://mirrors.aliyun.com/pypi/simple/
二、解决方案
得执行以下这行命令才可以**(不行的话,换个时间点再跑跑试试,或换个镜像)**
conda install pytorch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0 pytorch-cuda=11.7 -c pytorch -c nvidia
这篇关于解决方案:在autodl环境下安装torch被killed掉的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!