访问PyTorch官方网站(https://pytorch.org/),点击首页“Get Started”按钮进入安装页面。在页面中选择以下配置:
conda(推荐新手)或pip(灵活);点击“Download”按钮,官网会自动生成对应的安装命令(如conda install pytorch torchvision cudatoolkit=11.8 -c pytorch或pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118)。
清华大学提供了PyTorch的镜像站点,下载速度更快且稳定。
.tar.bz2文件(如pytorch-2.1.2-py3.10_cuda12.1_cudnn8.9.2_0.tar.bz2),右键复制链接地址,使用wget命令下载(如wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/pytorch-2.1.2-py3.10_cuda12.1_cudnn8.9.2_0.tar.bz2)。.whl文件(如torch-2.1.2+cu121-cp310-cp310-linux_x86_64.whl)。https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/pytorch-2.1.2-py3.10_cuda11.8_cudnn8.6.0_0.tar.bz2https://download.pytorch.org/whl/cu118/torch-2.1.2%2Bcu118-cp310-cp310-linux_x86_64.whlhttps://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/pytorch-2.1.2-py3.8_cpu_0.tar.bz2https://download.pytorch.org/whl/cpu/torch-2.1.2%2Bcpu-cp38-cp38-linux_x86_64.whlconda install --use-local 包名.tar.bz2安装,Wheel包用pip install 包名.whl安装。