pycharm 安装 pytorch 验证cuda

进入Pytorch官网
选择自己的配置,官网会提示两行命令,照着做就行了

pip3 install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp36-cp36m-linux_x86_64.whl
pip3 install https://download.pytorch.org/whl/cu100/torchvision-0.3.0-cp36-cp36m-linux_x86_64.whl

验证cuda

import torch

print(torch.cuda.is_available())
print(torch.cuda.device_count())
print(torch.cuda.get_device_name())
print(torch.cuda.current_device())

你可能感兴趣的:(习惯养成,开发环境配置,pytorch,Deep,Learning)