pip install pytorch or conda install pytorch

the hague

已于 2022-09-20 17:14:16 修改

阅读量1k

点赞数

分类专栏： python 目标检测文章标签： pytorch pip conda

于 2022-09-20 16:58:54 首次发布

本文链接：https://blog.csdn.net/percentagec/article/details/126957031

版权

目标检测同时被 2 个专栏收录

3 篇文章 0 订阅

订阅专栏

python

2 篇文章 0 订阅

订阅专栏

## 测试是否安装成功
import torch

## GPU
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())
print(torch.cuda_version)
print(torch.backends.cudnn.version())

## CPU
print(torch.__version__)

windows条件：CPU
conda install pytorch torchvision torchaudio cpuonly -c pytorch
pip3 install torch torchvision torchaudio

windows条件：CUDA11.3
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

windows条件：CUDA11.6
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

Linux条件：CPU
conda install pytorch torchvision torchaudio cpuonly -c pytorch
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu

Linux条件：CUDA10.2
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
pip3 install torch torchvision torchaudio

Linux条件：CUDA11.3
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

Linux条件：CUDA11.6
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116