问题三:安装pytorch-gpu版本时会默认安装cpu版本
大概就是通过在pytorch官网复制的指令安装pytorch时,明明安装的时GPU版本的,但是却是CPU版本的,卸载重新安装也是这样
conda install pytorch1.13.1 torchvision0.14.1 torchaudio==0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia
原因
不清楚
解决方案
使用pip指令安装
pip install torch1.13.1+cu117 torchvision0.14.1+cu117 torchaudio==0.13.1 -i https://pypi.tuna.tsinghua.edu.cn/simple --extra-index-url https://download.pytorch.org/whl/cu117
安装成功
标签:问题,https,遇到,cu117,学习,pytorch,版本,pip,安装 From: https://www.cnblogs.com/gs217/p/18424243