Win10+Anaconda,安装PyTorch、cudatoolkit、cudnn

Nina ·
更新时间:2024-11-10
· 621 次阅读

目录

为 PyTorch 创建虚拟环境

使用conda安装PyTorch (失败)

使用pip安装pytorch

安装cudatoolkit和cudnn

测试安装是否成功

首先,安装环境是:操作系统 Win10,已经预先安装了 Anaconda。关于 Anaconda 的安装步骤这里就忽略不讲了,Win10 下安装 Anaconda 非常简单。

为 PyTorch 创建虚拟环境

安装 Anaconda 完毕后,我们在安装 PyTorch 之前最好先创建一个 pytorch 的虚拟环境。之所以创建虚拟环境是因为 Python 为不同的项目需求创建不同的虚拟环境非常常见。在实际项目开发中,我们通常会根据自己的需求去下载各种相应的框架库,但是可能每个项目使用的框架库并不一样,或使用框架的版本不一样,这样需要我们根据需求不断的更新或卸载相应的库,管理起来相当麻烦。所以通过创建虚拟环境,相当于为不同的项目创建一块独立的空间,在这个空间里,你安装任何库和框架都是独立的,不会影响到外部环境。

总之就是想象成为 PyTorch 单独创建一个空间,这个空间与其它空间(如TensorFlow、其它版本的Python )互不影响。

创建 PyTorch 虚拟环境步骤:

1.打开 Anaconda 自带的 Anaconda Prompt:

2.打开 Anaconda Prompt 之后,在命令行输入:

conda create --name pytorch python=3.6

注意,这里的 pytorch 是虚拟环境的名称,可随意取。

3.然后继续下列命令,进入 pytorch 虚拟环境:

activate pytorch

效果如图所示:

好了,pytorch 的虚拟环境创建好了。接下来就是安装 PyTorch 了。

使用conda安装PyTorch (失败)

PyTorch 的官网提供了简单的安装方法,只需简单的命令即可。

首先,打开 PyTorch 官网安装页面:https://pytorch.org/get-started/locally/

我一开始是用的conda的方式安装,在下载pytorch1.4的安装包的时候1个小时都不动,一直是0%(如下图),遂放弃。

使用pip安装pytorch

然后我根据我自己的电脑环境,在pytorch官网中选择pip方式安装(如下图):

这样我们的pytorch就算安装好了。进入anaconda图形界面,可以看到在environment下,多了一个名叫“pytorch”的环境:

安装cudatoolkit和cudnn

首先看一下电脑CUDA的版本。控制面板——NVIDIA控制面板——帮助——系统信息——组件:

然后直接在anaconda的GUI中选择对应的版本依次安装即可:

测试安装是否成功 import torch print(torch.cuda.is_available())

返回结果是True,则PyTorch和CUDA安装成功

参考资料:https://www.jb51.net/article/155635.htm


作者:游离丸子



pytorch cudnn anaconda win

需要 登录 后方可回复, 如果你还没有账号请 注册新账号