pytorch安装详细步骤

文章目录

    • 环境配置步骤
    • 官网选择型号
    • 清华源快速安装
    • 安装完检验
    • pycharm设置项目环境

环境配置步骤

如果显卡支持GPU,在命令行处,输入nvidia-smi ,查看自己驱动的版本。
参考文章
pytorch安装详细步骤_第1张图片

官网选择型号

官网
pytorch安装详细步骤_第2张图片

清华源快速安装

参考博文

安装完检验

输入torch.cuda.is_available(),显示True则意味着可以使用GPU。
在这里插入图片描述

pycharm设置项目环境

pytorch安装详细步骤_第3张图片
这一步是容易忽略导致操作错误的地方,设置环境项目注意如果是导入已有的虚拟环境,需要选择conda environment,如果是新建虚拟环境,则选择virtualenv environment
这里因为需要导入已有的安装pytorch的环境,所以正确操作如下:
pytorch安装详细步骤_第4张图片

之后可以看到对应的编译器已经添加到:
pytorch安装详细步骤_第5张图片
项目显示编译器未设置,进行再次设置。
pytorch安装详细步骤_第6张图片

你可能感兴趣的:(deeplearning)