当前位置:   article > 正文

下载GPU版本Pytorch_下载gpu版本的pytorch

下载gpu版本的pytorch

参考小土堆视频下载

27. GPU版本-安装 PyTorch(方法2)_哔哩哔哩_bilibili

命令行查看本设备的cudaDriver版本

nvidia-smi

查看本驱动能够支持最大版本cuda:

在pytorch官网下载对应版本

打开cmd

conda activate pytorchName

方法一:直接安装

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

方法二:本地安装

复制下载地址,通过迅雷快速下载到本地

cmd下载

(pytorch) C:\Users\MyLENOVO>pip install address

下载结束后再次执行

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

下载剩余的包

下载成功即可完成

验证:

进入pytorch环境,

  1. (pytorch) C:\Users\LENOVO>python
  2. >>> import torch
  3. >>> torch.cuda.is_available()
  4. True

完成

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/IT小白/article/detail/975367
推荐阅读
相关标签
  

闽ICP备14008679号