Window下Python+CUDA+PyTorch怎么安装
这篇文章主要讲解了“Window下Python+CUDA+PyTorch怎么安装”,文中的讲解内容简单清晰,易于学习与理解,下面请大家跟着小编的思路慢慢深入,一起来研究和学习“Window下Python+CUDA+PyTorch怎么安装”吧!
公司主营业务:网站制作、成都网站制作、移动网站开发等业务。帮助企业客户真正实现互联网宣传,提高企业的竞争能力。创新互联建站是一支青春激扬、勤奋敬业、活力青春激扬、勤奋敬业、活力澎湃、和谐高效的团队。公司秉承以“开放、自由、严谨、自律”为核心的企业文化,感谢他们对我们的高要求,感谢他们从不同领域给我们带来的挑战,让我们激情的团队有机会用头脑与智慧不断的给客户带来惊喜。创新互联建站推出章丘免费做网站回馈大家。
1 概述
Windows
下Python+CUDA+PyTorch
安装,步骤都很详细,特此记录下来,帮助读者少走弯路。
2 Python
Python
的安装还是比较简单的,从官网下载exe
安装包即可:
因为目前最新的torch
版本只支持到Python 3.8
,因此为了不会出现版本兼容问题建议安装Python 3.8
。
下载中间的那个executable installer
并安装,打开会建议选择自定义:
选择需要的组件:
选择位置:
安装完后可以从命令行查看是否安装成功(一般默认会把路径添加到环境变量中):
python --version
由于该安装包会自动把pip
也一起安装,因此在使用时会提示pip
需要升级,请使用如下命令升级pip
:
python -m pip install --upgrade pip
在安装前建议添加国内源,在用户目录下新建一个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
:
下载好后打开(实际上是一个压缩包,这里是解压路径):
解压完成后会自动打开安装界面:
没特殊需求可以直接使用默认安装选项,否则请选择自定义组件进行安装:
由于笔者没有安装VS
,这里会提示找不到VS
,忽略即可:
接下来就是等待安装:
安装完成:
打开cmd
输入nvcc --version
,显示版本号即为安装成功:
4 PyTorch
4.1 下载
PyTorch
建议使用离线方式手动安装,戳这里下载对应CUDA
版本与对应Python
版本的torch
以及torchvision
:
最前面的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
注意如果是Windows 2004
版本可能会出现如下问题:
这是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怎么安装
网页链接:http://scyanting.com/article/igodpd.html