前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
社区首页 >专栏 >pytorch:不同版本的CUDA和CUDNN安装教程

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

作者头像
AI拉呱
发布于 2021-01-14 02:19:16
发布于 2021-01-14 02:19:16
2.1K00
代码可运行
举报
运行总次数:0
代码可运行

通过Anaconda 安装

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

具体如下

cuda9.0

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

CUDA 8.x

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
conda install pytorch torchvision cudatoolkit=8.0 -c pytorch

CUDA 10.0

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

通过pip安装

python3

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
# 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

参考官网

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2019/03/21 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档
本文部分代码块支持一键运行,欢迎体验
本文部分代码块支持一键运行,欢迎体验