问题描述:
刚接触pytorch,在pycharm中用pip安装的pytorch,在使用过程中发现没有智能提示的问题(已解决)
项目中pytorch的包能成功引入并运行,但在pycharm的环境中找不到pytorch包,如图
经过查询发现没有提示就是因为pycharm的环境中没有pytorch包(目前还不知道为什么pip安装后在pycharm中没有)
问题解决:
用anaconda安装pytorch
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
其中的10.2是安装的pytorch的版本号,要根据你自己的CUDA去选择
安装成功后在pycharm将编译器改成anaconda的
点击ADD
选择Conda Environment,选择Existing environment,选择自己的anaconda安装路径和pytorch的安装路径,点击ok保存
如果package中没有pytorch包的话需要手动导入,点击+号,新弹出的页面点击Manage Respositories
选择pytorch,并点击ok保存
在新弹出的页面中找到pytorch,点击后选择install Package即可成功在环境中添加pytorch包,最后记得Apply并保存
问题解决,出现pytorch的智能提示