一、现象
平台: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环境,发现执行下面的命令行之后,也显示安装成功了,但在jupyter notebook打印版本号发现对应不上?
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/
二、解决方案
安装所处的环境不同所导致!!!
需要切换对应的环境(source activate、conda activate 环境名),在终端上执行,文件转化为.py
(tips:遇见被killed的话,换个时间点再跑跑试试,或换个镜像)
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