pytorch下载指令
时间: 2023-11-16 09:56:57 浏览: 96
pytorch清华源下载.zip
PyTorch是一个开源的机器学习框架,可以用于构建深度神经网络。以下是PyTorch下载指令:
1. 下载CPU版本:pip install torch torchvision torchaudio
2. 下载GPU版本:pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu102/torch_stable.html
3. 下载指定版本:pip install torch==1.8.1+cu102 torchvision==0.9.1+cu102 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
请注意,第三个指令中的版本号可以根据需要进行更改。同时,为了提高下载速度,可以添加-i参数指定清华镜像源,例如:-i https://pypi.tuna.tsinghua.edu.cn/simple。
如果您想检查PyTorch是否安装成功并且是否可以使用GPU,可以在Python中输入以下命令:
import torch
print(torch.cuda.is_available())
如果输出结果为True,则说明PyTorch已经成功安装并且可以使用GPU。
阅读全文