首先我的电脑之前装有pytorch,但是最近想用pycharm,因此装了pycharm,但是碰到了一系列的问题:
1.设置torch环境,import torch 返回no module named “torch”
2.解决第一个问题后,出现了torch.cuda.is_available()返回false的问题
接下来我来给出着俩问题的解决办法

1.设置torch环境,import torch 返回no module named “torch”

查阅各种解决setting环境的办法,无果,不能解决我这里的问题

我参考的博文import torch显示No module named ‘torch‘在pycharm中

相应的解决办法我也写在这里

首先电脑里装有torch和torchvision
1.重新建一个项目
2.选择安装pytorch虚拟环境下的python
3.然后导入torch,就成功了


尝试过各种建好project后setting project的环境,但是都以失败而告终、、、

2.torch.cuda.is_available()返回false的问题

然而我万万没想到的是,导入torch成功了,torch.cuda.is_available()居然返回false,在此之前cmd里一直返回true的,我至今不知道是怎么一回事、、、望各位大佬指教

说回解决办法,既然问题出现了,以上疑问就先放一边,先解决了问题再说。

继续网上各种查阅,大家给出的原因都是:cuda版本和torch版本不匹配

先去cmd窗口查看cuda版本,cmd窗口输入命令:nvcc -V (注意V事大写)

我的torch版本不知道什么时候被我自己升级到了1.4.0了,torchvision也不能用了、、、

怎么办?要么重装cuda,要么重装torch,要么俩都重新装。由于我的情况事torchvision也坏掉了,所以就决定重装torch,不行再重装cuda、、、、

1.由于我之前是装有pytorch虚拟环境的,所以第一步要先删除虚拟环境,这里我选择的删除虚拟环境,没有用pip uninstall torch,因为删除虚拟环境比较干净,况且我的torchvision也坏掉了、、、

打开Anaconda prompt,输入命令:conda env remove --name [env-name]
比如我的虚拟环境叫pytorch,所以我输入命令:
conda env remove --name pytorch

2.创建虚拟环境,创建虚拟环境的好处在于可以让你的环境干净,不乱,强烈建议

在Anaconda prompt中,输入conda -n pytorch python=3.6
其中pytorch是虚拟环境的名字,python=3.6是python的版本

3.激活虚拟环境
我还是在Anaconda prompt中操作的,也可以在cmd中操作
输入activate pytorch,其中pytorch是虚拟环境的名字。

4.安装pytorch
同在Anaconda prompt中操作,我装的torch是1.1.0,torchvision版本是0.3.0
先装torch,命令是(官网找哈)
pip install https://download.pytorch/whl/cu100/torch-1.1.0-cp36-cp36m-win_amd64.whl
再装torchvision,命令
pip install https://download.pytorch/whl/cu100/torchvision-0.3.0-cp36-cp36m-win_amd64.whl

几点说明

  1. 可以用镜像来装,官网下载的话早上网速很快,所以也可以官网早上装
  2. **官网给的是用pip3 install…**在创建虚拟环境的时候,会给出你的配置是pip还是pip3,如果是pip,就用我上面的命令pip install…

    5.安装成功测试

因为成功了,也就没有重装cuda

总结:

  1. torch没有导入成功就是环境没有配置成功,可以在file——setting——project——interpreter中设置,不行的话就新建project的时候设置成pytorch中的python解释器就好了
  2. 装有cuda,但是cuda.is_available()返回false,那就是版本不对应,重新安装pytorch省事
  3. 虚拟环境好用,可以干净的卸载

祝大家好运

更多推荐

pycharm不能import torch问题+import torch 成功后,torch.cuda.is_available() false问题解决