环境配置步骤
如果显卡支持GPU,在命令行处,输入nvidia-smi
,查看自己驱动的版本。
参考文章
官网选择型号
清华源快速安装
安装完检验
输入torch.cuda.is_available()
,显示True则意味着可以使用GPU。
pycharm设置项目环境
这一步是容易忽略导致操作错误的地方,设置环境项目注意如果是导入已有的虚拟环境,需要选择conda environment
,如果是新建虚拟环境,则选择virtualenv environment
。
这里因为需要导入已有的安装pytorch的环境,所以正确操作如下:
之后可以看到对应的编译器已经添加到:
项目显示编译器未设置,进行再次设置。