pytorch-gpu安装-win10-python3.7-anaconda3

文章目录

    • 1.安装
    • 2.测试

1.安装

创建虚拟环境pytorch,pip 安装gpu版本,没有cuda就选安装cpu

conda create -n pytorh python=3.7
pip install torch==1.7.0+cu101 torchvision==0.8.1+cu101 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

https://pytorch.org/
pytorch-gpu安装-win10-python3.7-anaconda3_第1张图片

2.测试

import torch    # 如正常则静默
a = torch.Tensor([1.])    # 如正常则静默
a.cuda()    # 如正常则返回"tensor([ 1.], device='cuda:0')"
from torch.backends import cudnn # 如正常则静默
cudnn.is_acceptable(a.cuda())    # 如正常则返回 "True"

pytorch-gpu安装-win10-python3.7-anaconda3_第2张图片

你可能感兴趣的:(IT,pytorch,深度学习,python)