深度学习--基础(一)pytorch安装--cpu & GPU版本

CPU版本安装

直接在线安装

无GPU的时候,只能安装CPU版本,打开官网 https://pytorch.org/ 直接Pip安装即可

国内访问这些下载安装会出现超时的情况,可以-i指定国内安装源:
pip3.11 install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple/
在这里插入图片描述
ps:python版本3.11 ,3.7版本安装时,遇到了如下报错osError找不到asmjit.dll的报错,升级3.11即可

离线安装

从这里下载 https://download.pytorch.org/whl/torch_stable.html
注意torch torchvision和 python版本是有对应关系,总结了一下:
在这里插入图片描述
下载下来后,可能会出现
whl is not a supported wheel on this platform.的报错
在这里插入图片描述
此时可以查看
pip debug --verbose
名字必须符合如下:
在这里插入图片描述
可以修改whi文件名安装即可。(有些64系统尾缀为win32,实际指的是win32接口,实质还是64位,总之修改成符合当前要求的命名规则的whl即可)

pip3.11 install torchvision-0.11.3+cpu-cp311-cp37m-amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple/
pip3.11 install torch-1.10.2+cpu-cp311-cp311m-amd64.whl

卸载

pip uninstall torchvision -y
pip uninstall torch -y

GPU版本安装

安装anaconda

这个是直接官网下载即可,是要给包和环境的管理器,可以不安装,如果没有安装anaconda,下面其他各步都可以可以使用pip,完全不影响使用,那么安装这个的目的是什么呢?

它可以创建不同的虚拟环境,防止与你本机或者其他环境的python发生冲突

anaconda需要先创建虚拟环境
conda create -n test
并且进入环境conda activate test
然后继续安装后面的python pytorch等

安装CUDA

GPU开发,这个必须安装,相当于sdk,window下下载安装,下载地址:
https://developer.nvidia.com/cuda-downloads
在这里插入图片描述
注意版本号必须是与显卡匹配的CUDA版本,CUDA版本号在下载文件名中这个位置
在这里插入图片描述

那么如何确认显卡匹配的CUDA版本?

  1. 显卡驱动的版本,使用NVIDIA-smi指令:
    在这里插入图片描述
    比如我这里显示518.01

  2. 确认显卡驱动可以对应的cuda版本,可以从下面这个连接查询:
    https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#major-components
    在这里插入图片描述
    我这里的518可以安装很多版本,我选择的是CUDA 11.7,接下来一步步傻瓜式安装 即可

安装pytorch

如果conda环境的,先进入环境再继续下载
进入官网 https://pytorch.org/
在这里插入图片描述
注意此处的CUDA选你安装的cuda 版本

另外可以从这里找以前的版本的下载方式:
https://pytorch.org/get-started/previous-versions/
在这里插入图片描述

安装后,在用NVIDIA-smi查看,显示了驱动和cuda版本:

在这里插入图片描述
都安装完毕后,进入python环境(anaconda的需要先进入虚拟环境)直接python
然后 import torch
如果不报错则是安装成功
而GPU版本,还需要输入下torch.cuda.is_available(),如果返回为True,则gpu版本安装成功

**注意:pytorch的版本对python的版本要求,如果报错版本冲突,可以卸载安装更高版本的python,conda update python
**

  • 5
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
### 回答1: PyTorch-GPU与CUDA版本对应如下: PyTorch-GPU 1..:CUDA 9. PyTorch-GPU 1.1.:CUDA 9.、CUDA 10. PyTorch-GPU 1.2.:CUDA 9.、CUDA 10.、CUDA 10.1 PyTorch-GPU 1.3.:CUDA 9.2、CUDA 10.、CUDA 10.1 PyTorch-GPU 1.4.:CUDA 10.、CUDA 10.1、CUDA 10.2 PyTorch-GPU 1.5.:CUDA 10.1、CUDA 10.2 PyTorch-GPU 1.6.:CUDA 10.1、CUDA 10.2、CUDA 11. PyTorch-GPU 1.7.:CUDA 10.1、CUDA 10.2、CUDA 11. PyTorch-GPU 1.8.:CUDA 11.1 PyTorch-GPU 1.9.:CUDA 11.1、CUDA 11.2 需要注意的是,不同版本PyTorch-GPU可能需要不同版本的CUDA才能正常运行。因此,在安装PyTorch-GPU时,需要根据自己的CUDA版本选择相应的PyTorch-GPU版本。 ### 回答2: PyTorch是一个流行的深度学习库,支持使用GPU加速算法运行以提高训练速度。在PyTorch中,CUDA是一种用于在NVIDIA GPU上加速计算的并行计算平台和API集合。因此,PyTorchGPU功能需要与CUDA版本兼容。 PyTorchGPU支持是通过与CUDA库进行交互来实现的。由于PyTorch和CUDA的版本兼容性问题,所以要使用GPU功能,需要确保安装有与PyTorch版本兼容的CUDA库。常见的PyTorch版本与CUDA版本对应关系如下: - PyTorch 1.0.x 对应 CUDA 9.0 - PyTorch 1.1.x-1.2.x 对应 CUDA 10.0 - PyTorch 1.3.x 对应 CUDA 10.1 - PyTorch 1.4.x-1.5.x 对应 CUDA 10.2 需要注意的是,不同的PyTorch版本和不同的GPU型号可能有不同的CUDA版本要求。因此,在使用GPU加速算法时,需要根据具体情况选择合适的PyTorch版本和CUDA版本。 总之,为了确保PyTorch能够充分利用GPU的加速能力,需要安装正确版本的CUDA库,并确保与PyTorch版本兼容。通过此功能使用GPU加速可以加快训练速度,提高模型性能。 ### 回答3: PyTorch是一个基于Python的开源机器学习框架,具有广泛的应用和活跃的社区支持。PyTorch可以在CPUGPU上运行,而使用GPU的最简单方法是使用CUDA(Compute Unified Device Architecture)工具包。CUDA是由NVIDIA开发的并行计算平台和应用程序接口,用于在GPU上进行高性能计算。因此,PyTorch需要与正确版本的CUDA协同工作,以在GPU上实现最佳性能。 不同版本PyTorch对应着不同版本的CUDA,因此在安装PyTorch之前需要确定使用哪个版本的CUDA。在PyTorch官网上推荐使用的版本如下: PyTorch 1.7.x:CUDA 10.1,CUDA 10.2或CUDA 11.0 PyTorch 1.6.x:CUDA 10.1,CUDA 10.2或CUDA 11.0 PyTorch 1.5.x:CUDA 10.1或CUDA 10.2 PyTorch 1.4.x:CUDA 10.1 PyTorch 1.3.x:CUDA 10.1 PyTorch 1.2.x:CUDA 9.2 需要注意的是,不同版本的CUDA需要特定的GPU架构才能运行,因此在安装CUDA之前,需要先了解自己的GPU支持哪些架构。此外,为了避免因为版本不匹配而发生不必要的打扰或错误,建议在安装PyTorch之前也检查一下自己机器上CUDA的版本。如果是多个版本共存,则需要设置环境变量以指示使用哪个版本。 总之,在安装和使用PyTorch时,需要对应选择正确的CUDA版本,以保证在GPU上获得最佳性能和稳定性。同时,需要了解自己机器上GPU的性能和支持的CUDA版本,以避免不必要的打扰和错误。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值