ubuntu 18.04 安装pytorch

环境:ubuntu 18.04 , CUDA9.0 , Python3.5

 去官网选择需要安装的对应版本,如果是想安装最新版本,直接在终端输入:

pip3 install torch torchvision

如果想安装其他版本的PyTorch,点击install previous versions of PyTorch选择需要安装的对应版本,然后下载torch-0.4.0-cp35-cp35m-linux_x86_64.whl(此示例对应python3.5,cuda9.0的PyTorch0.4.0版本),下载完成后,直接执行下面命令即可

pip3 install torch-0.4.0-cp35-cp35m-linux_x86_64.whl

 

发布了30 篇原创文章 · 获赞 22 · 访问量 3万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览