pytorch的安装

本文介绍了如何在CUDA10.0环境下安装匹配的PyTorch版本。强调了CUDA版本与PyTorch版本之间的兼容性,并提供了conda安装命令。注意,PyTorch官网未直接提供CUDA10.0的版本,需要使用指定命令进行安装。
摘要由CSDN通过智能技术生成
  • 首先必须要安装好cuda和cudann,我安装的是如下版本的cuda
  •  需要注意cuda版本和之后的pytorch版本需要匹配
  • cuda10.0环境安装Pytorch_先说一下,这里不是博客,只是我的笔记本-CSDN博客_cuda10.0对应的pytorch
  • # CUDA 9.0
    conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch

    # CUDA 10.0
    conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorch

    # CPU Only
    conda install pytorch-cpu==1.1.0 torchvision-cpu==0.3.0 cpuonly -c pytorch

  • 由于我选择的是cuda10.0,所以安装第二个版本的Pytorch

  • Pytorch官网有以下版本,但是没有cuda10.0,因此只能用上面的来安

  •  

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值