在Pycharm中加载cuda,创建pytorch环境

前言

在Pycharm中使用Anaconda创建Pytorch虚拟环境

报错的内容

分析问题

解决方法

PyTorch与Torchvision对应关系

后序

在Anaconda中写代码的时候,有时候需要写很多的代码,并且还需要导入很多的库,因此在Jupyter Notebook中写代码会有很多麻烦的事情,所以在Pycharm这个工具中写起来是很方便的。

在Pycharm中使用Anaconda创建Pytorch的虚拟环境

  • 如下图标记所示,打开Pycharmsettings,修改Project的编译器。或者是在创建新的环境的时,选择Anaconda的创建的Pytorch-->  python的虚拟环境。在Pycharm中加载cuda,创建pytorch环境_第1张图片

在选择好编译的环境后,然后在Pycharm的命令输出窗口(Terminal)中输入 import torch, import numpy 时会发生错误。

报错的内容

当我们输入 import torch 时,会发生

Please note and check the following:

  * The Python version is: Python3.8 from "D:\anacondafile\envs\qr_env\python.exe"
  * The NumPy version is: "1.21.5"

and make sure that they are the versions you expect.
Please carefully study the documentation linked above for further help.

Original error was: DLL load failed: 找不到指定的模块。
 

当输入 import torchvision 时会发生:

Traceback (most recent call last):
  File "D:/qr_project/encoder/test.py", line 1, in
    import torchvision
  File "D:\anacondafile\envs\qr_env\lib\site-packages\torchvision\__init__.py", line 5, in
    from torchvision import models
  File "D:\anacondafile\envs\qr_env\lib\site-packages\torchvision\models\__init__.py", line 1, in
    from .alexnet import *
  File "D:\anacondafile\envs\qr_env\lib\site-packages\torchvision\models\alexnet.py", line 1, in
    import torch
  File "D:\anacondafile\envs\qr_env\lib\site-packages\torch\__init__.py", line 189, in
    from torch._C import *
UnicodeDecodeError: 'utf-8' codec can't decode byte 0xe0 in position 0: invalid continuation byte
 

分析问题

值得注意的是,此项目通过Anaconda创建自己的环境,并且安装了torch, torchvision等工具包,在Anaconda Prompt的命令窗口是可以正常导入的,但是在Pycharm编译器中会发生错误!

通过activate PyTorch 进入到自己创建的 Pytoch 环境中,输入python会出现对应的python版本信息,在输入import ,是不会报错的。

这是因为在Anaconda的环境下有支持import torch 的 dll ,然而在Pycharm中使用的是创建的新环境,还缺少相应支持的dll, 因此仅需要在Pycharm中配置环境变量

解决方法

1.首先在Pycharm的主界面点击编辑器Run->Edit Configurations 就会出现如下图所示的界面,添加环境变量,然后点击右边的选项。

在Pycharm中加载cuda,创建pytorch环境_第2张图片

在Pycharm中加载cuda,创建pytorch环境_第3张图片 

2.点击 + ,新建环境变量 PATH :

 在Pycharm中加载cuda,创建pytorch环境_第4张图片

 然后就是设置 Value 

D:\Anacondafile;
D:\Anacondafile\Library\mingw-w64\bin;
D:\Anacondafile\Library;
D:\Anacondafile\Lib\site-packages;
D:\Anacondafile\Scripts;
D:\Anacondafile\envs\PyTorch\Library\bin;

具体的路径,根据你Anaconda的安装目录自行调整。 

 其中如果出现 import torchvison 不成功的问题;然后复制一下【image.pyd】,利用该文件在创建一个副本 【image.py】 ,更改之后在导入 import torchvision 就可以成功了

在Pycharm中加载cuda,创建pytorch环境_第5张图片

 在Pycharm中加载cuda,创建pytorch环境_第6张图片

 因为在做分割模型,检测模型等会需要到torchvision, 因此这个是很重要的。

后序

起航

你可能感兴趣的:(pytorch,pycharm,python)