PyTorch 概述
教程
安装
相关链接:
conda create --name torch
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch
pip install tensorflow
使用 pip 安装的时候必须注意 CUDA 版本,直接 pip install pytorch
安装的版本没有 CUDA 支持。
在这里找支持你的 CUDA 的 PyTorch 版本:https://pytorch.org/get-started/previous-versions/
pip install torch==1.7.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html
验证安装是否成功(使用 IPython 的话要确保当前环境里有安装 IPython):
import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.get_device_name(0))
推荐阅读
Links: pytorch-abstract