PyTorch 概述

    
        标签：
        
            PyTorch 相关笔记
        
    发布于：2021-05-21 12:37:39
    编辑于：2021-05-21 12:37:39
    浏览量：1195

教程

相关链接：

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))