欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Ubuntu16.04安装pytorch方法

程序员文章站 2022-05-13 12:47:34
Ubuntu16.04 Anaconda3安装pytorch方法准备工作安装cuda10.0及cudnn7.4.下载pytorch安装文件安装pytorch准备工作安装cuda10.0及cudnn7.4.请参考我这篇文章:Ubuntu16.04安装cuda及cudnn方法下载pytorch安装文件torch官网上可找到各种安装方式,这里选择whl安装,首先在网站上下载对应cuda10.0版本的whl文件。这里我下载的是1.0.1.post2(一个项目指定要用这个版本,:<我也没办法),又由于...

准备工作

安装cuda10.0及cudnn7.4.

请参考我这篇文章:Ubuntu16.04安装cuda及cudnn方法

下载pytorch安装文件

torch官网上可找到各种安装方式,这里选择whl安装,首先在网站上下载对应cuda10.0版本的whl文件。这里我下载的是1.0.1.post2(一个项目指定要用这个版本,:<我也没办法),又由于我用的python3.5,因此选择下载torch-1.0.1.post2-cp35-cp35m-linux_x86_64.whl这个安装。

安装pytorch

sudo chmod a+x torch-1.0.1.post2-cp35-cp35m-linux_x86_64.whl
pip install torch-1.0.1.post2-cp35-cp35m-linux_x86_64.whl

另外还需安装torchvision,这里对应的版本是0.2.2.post2,用pip安装:

pip install torchvision==0.2.2.post2 -i https://pypi.tuna.tsinghua.edu.cn/simple/

测试是否安装成功:

import torch
print(torch.__version__)           #输出Pytorch的版本
print(torch.cuda.is_available())   #看看能否正常调用CUDA

Ubuntu16.04安装pytorch方法
一切正常,大功告成。

本文地址:https://blog.csdn.net/wolfcsharp/article/details/107474382