首页 > 系统服务 > 详细

Linux下使用anaconda安装pytorch

时间:2020-02-24 21:42:52      阅读:99      评论:0      收藏:0      [点我收藏+]

添加清华镜像(下载速度快):

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

安装指定版本的pytorch(cuda版本为9.0):

conda install pytorch=1.0.0 torchvision cuda90

安装完成:

技术分享图片

 进入python环境测试pytorch是否安装成功:

>>> import torch
>>> print(torch.__version__)

技术分享图片

 安装成功。这样就可以使用pytorch啦!

技术分享图片

 

Linux下使用anaconda安装pytorch

原文:https://www.cnblogs.com/douliyoutang01/p/12358902.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!