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

AI拉呱

发布于 2021-01-14 10:19:16

2.5K0

通过Anaconda 安装

具体如下

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

conda install pytorch torchvision cudatoolkit=8.0 -c pytorch

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

# 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 条评论

热度