pytorch:不同版本的CUDA和CUDNN安装教程

本文提供了使用Anaconda和pip在不同CUDA版本(9.0, 8.x, 10.0)下安装PyTorch的详细教程,特别强调了PyTorch安装依赖于CUDA的具体版本,并给出了Python3环境下通过pip安装的方法。" 108761289,158351,使用.NET Core构建Agora AccessToken服务,"['C#', 'ASP.NET Core', '视频通信', '安全认证', '微服务']
摘要由CSDN通过智能技术生成

通过Anaconda 安装

pytorch 是根据不同的cuda版本安装的

具体如下

cuda9.0

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

CUDA 8.x

conda install pytorch torchvision cudatoolkit=8.0 -c pytorch

CUDA 10.0

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

通过pip安装

python3

# Python 3.5
pip3 install https://download.pytorch.org/whl/cpu/torch-1.0.1.post2-cp35-cp35m-linux_x86_64.whl
pip3 install torchvision

参考官网

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

AI拉呱

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值