pytorch安装详细步骤

环境配置步骤

如果显卡支持GPU,在命令行处,输入nvidia-smi ,查看自己驱动的版本。
参考文章
在这里插入图片描述

官网选择型号

官网
在这里插入图片描述

清华源快速安装

参考博文

安装完检验

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

pycharm设置项目环境

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

之后可以看到对应的编译器已经添加到:
在这里插入图片描述
项目显示编译器未设置,进行再次设置。
在这里插入图片描述

发布了5 篇原创文章 · 获赞 1 · 访问量 90

猜你喜欢

转载自blog.csdn.net/spatial_coder/article/details/104864396