一分钟下载完pytorch

-i https://pypi.tuna.tsinghua.edu.cn/simple

pip install torch==2.3.1 --index-url https://download.pytorch.org/whl/cu118

找网址之前就不说了 

可以参考

CUDA卸载&&重装-CSDN博客

【深度学习:入门】如何配置CUDA,使用gpu本地训练_cuda怎么使用_氯化氯的博客-CSDN博客

先总结一下之前容易犯的错误

1.下载cuda的时候别改下载地址

2.自己配置上环境变量,怎么配置可以参考第一个链接

这篇博客的重点是将怎么用国内镜像下载pytorch

首先在pytorch获得对应的下载链接

能用pip就不要conda

能用pip就不要conda

能用pip就不要conda

无论你是在cmd或者anaconda或者哪里按 命令行能用pip别用conda

conda install pytorch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0 pytorch-cuda=11.6 -c pytorch -c nvidia

这个conda命令看不懂

pip install torch==1.13.0+cu116 torchvision==0.14.0+cu116 torchaudio==0.13.0 --extra-index-url https://download.pytorch.org/whl/cu116

pip命令就简单了后面的网址乐意改成啥改成啥

可以参考CUDA11.6的pytorch环境搭建 - CSDN文库

我用到的源是-f https://download.pytorch.org/whl/torch_stable.html

用的时候

pip install torch==1.13.0+cu116 torchvision==0.14.0+cu116 torchaudio==0.13.0 -f https://download.pytorch.org/whl/torch_stable.html

两分钟基本就安装好了

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐