温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

Window下Python+CUDA+PyTorch怎么安装

发布时间:2021-11-23 18:05:23 来源:亿速云 阅读:161 作者:iii 栏目:大数据

这篇文章主要讲解了“Window下Python+CUDA+PyTorch怎么安装”,文中的讲解内容简单清晰,易于学习与理解,下面请大家跟着小编的思路慢慢深入,一起来研究和学习“Window下Python+CUDA+PyTorch怎么安装”吧!

1 概述

WindowsPython+CUDA+PyTorch安装,步骤都很详细,特此记录下来,帮助读者少走弯路。

2 Python

Python的安装还是比较简单的,从官网下载exe安装包即可:

Window下Python+CUDA+PyTorch怎么安装

因为目前最新的torch版本只支持到Python 3.8,因此为了不会出现版本兼容问题建议安装Python 3.8

下载中间的那个executable installer并安装,打开会建议选择自定义:

Window下Python+CUDA+PyTorch怎么安装

选择需要的组件:

Window下Python+CUDA+PyTorch怎么安装

选择位置:

Window下Python+CUDA+PyTorch怎么安装

Window下Python+CUDA+PyTorch怎么安装

安装完后可以从命令行查看是否安装成功(一般默认会把路径添加到环境变量中):

python --version

Window下Python+CUDA+PyTorch怎么安装

由于该安装包会自动把pip也一起安装,因此在使用时会提示pip需要升级,请使用如下命令升级pip

python -m pip install --upgrade pip

Window下Python+CUDA+PyTorch怎么安装

在安装前建议添加国内源,在用户目录下新建一个pip文件夹,并在其中新建一个叫pip.ini的文件,输入如下内容:

[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
[install]
trusted-host = https://pypi.tuna.tsinghua.edu.cn

3 CUDA

CUDA的安装也不难,一般Windows会在装系统的时候自动安装好对应的N卡驱动,从官网下载对应的CUDA Toolkit

Window下Python+CUDA+PyTorch怎么安装

下载好后打开(实际上是一个压缩包,这里是解压路径):

Window下Python+CUDA+PyTorch怎么安装

Window下Python+CUDA+PyTorch怎么安装

解压完成后会自动打开安装界面:

Window下Python+CUDA+PyTorch怎么安装

没特殊需求可以直接使用默认安装选项,否则请选择自定义组件进行安装:

Window下Python+CUDA+PyTorch怎么安装

由于笔者没有安装VS,这里会提示找不到VS,忽略即可:

Window下Python+CUDA+PyTorch怎么安装

接下来就是等待安装:

Window下Python+CUDA+PyTorch怎么安装

Window下Python+CUDA+PyTorch怎么安装

安装完成:

Window下Python+CUDA+PyTorch怎么安装

打开cmd输入nvcc --version,显示版本号即为安装成功:

Window下Python+CUDA+PyTorch怎么安装

4 PyTorch

4.1 下载

PyTorch建议使用离线方式手动安装,戳这里下载对应CUDA版本与对应Python版本的torch以及torchvision

Window下Python+CUDA+PyTorch怎么安装

最前面的cu110表示cuda 11.0,接着是torch/torchvision,再接着是版本,后面的cp表示Python版本,比如cp36表示Python 3.6,接着是系统和架构。

4.2 安装

下载完成后使用pip安装即可:

pip install torch-1.7.0+cu110-cp38-cp38-win_amd64.whl
pip install torchvision-0.8.1+cu110-cp38-cp38-win_amd64.whl

Window下Python+CUDA+PyTorch怎么安装

注意如果是Windows 2004版本可能会出现如下问题:

Window下Python+CUDA+PyTorch怎么安装

这是numpy 19.0.4版本的问题,详情可以戳这里。

解决方法是安装19.0.3版本:

pip uninstall numpy
pip install numpy==19.0.3

接着进行测试,打开python

python
import torch
print(torch.__version__)

输出torch版本即表示成功:

Window下Python+CUDA+PyTorch怎么安装

感谢各位的阅读,以上就是“Window下Python+CUDA+PyTorch怎么安装”的内容了,经过本文的学习后,相信大家对Window下Python+CUDA+PyTorch怎么安装这一问题有了更深刻的体会,具体使用情况还需要大家实践验证。这里是亿速云,小编将为大家推送更多相关知识点的文章,欢迎关注!

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI