这些天安装 pytorch,遇到了一些坑,特此总结一下,以免忘记。分享给大家。
首先,安装环境是:操作系统 win10,已经预先暗转了 anaconda。
1. 为 pytorch 创建虚拟环境
关于 anaconda 的安装步骤这里就忽略不讲了,win10 下安装 anaconda 非常简单。
安装 anaconda 完毕后,我们在安装 pytorch 之前最好先创建一个 pytorch 的虚拟环境。之所以创建虚拟环境是因为 python 为不同的项目需求创建不同的虚拟环境非常常见。在实际项目开发中,我们通常会根据自己的需求去下载各种相应的框架库,但是可能每个项目使用的框架库并不一样,或使用框架的版本不一样,这样需要我们根据需求不断的更新或卸载相应的库,管理起来相当麻烦。所以通过创建虚拟环境,相当于为不同的项目创建一块独立的空间,在这个空间里,你安装任何库和框架都是独立的,不会影响到外部环境。
总之就是想象成为 pytorch 单独创建一个空间,这个空间与其它空间互不影响。我之前安装了 tensorflow 了,也是创建了一个虚拟环境,且 python 版本是 3.5 的。所以,这里当然也要创建虚拟环境了。
创建 pytorch 虚拟环境步骤:
1.打开 anaconda 自带的 anaconda prompt:
2.打开 anaconda prompt 之后,在命令行输入:
1
|
conda create - - name pytorch python = 3.6 |
注意,这里的 pytorch 是虚拟环境的名称,可随意取。
3.然后继续下列命令,进入 pytorch 虚拟环境:
1
|
activate pytorch |
好了,pytorch 的虚拟环境创建好了。接下来就是安装 pytorch 了。
其实,安装 pytorch 还是非常简单的。接下来记录一些坑。
2. pytorch 官网安装
pytorch 的官网提供了简单的安装方法,只需简单的命令即可。
首先,打开 pytorch 官网安装页面(需自备梯子):https://pytorch.org/get-started/locally/
可以看到页面里的这张图,非常简单。我们的环境是:windows、conda、python 3.6、none cuda,对应的命令是:
1
|
conda install pytorch - cpu torchvision - cpu - c pytorch |
复制这条命令在 anaconda prompt 里输入,则 pytorch “顺利”安装完成。
但是问题来了,在 jupyter 里 import torch
没有问题,可 import torchvision
确报错了:
importerror: dll load failed: 找不到指定的模块。
可能是 pillow
模块有问题。尝试卸载 pillow
模块,但是会同时卸载 pytorch,故为一坑。
3. pip 安装
上一种方法舍弃,这里尝试使用 pip 安装。
同样在 anaconda prompt 里输入以下命令:
1
2
3
|
# python 3.6 pip3 install https: / / download.pytorch.org / whl / cpu / torch - 1.0 . 0 - cp36 - cp36m - win_amd64.whl pip3 install torchvision |
安装顺利完成。
这时候在 jupyter 里 import torch
没有问题,可 import torchvision
确仍然报错了:
importerror: dll load failed: 找不到指定的模块。
解决方案:
在 anaconda prompt 里输入以下命令:
1
2
|
conda uninstall pillow pip install pillow |
使用 pip 重新安装 pillow
模块即可。
至此,填坑。pytorch 成功安装。以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持服务器之家。
原文链接:https://blog.csdn.net/red_stone1/article/details/86669362