ubuntu16.04 gtx1080ti 安装pytorch

         最近想用pythorch训练yolo,所以搭建了一下pytorch的开发环境

        pytorch安装虽然简单但是也是有很多坑的!首先我选择的是cuda9.0版本的,我的显卡驱动是nvida-384.130.安装方法在我上一篇博客中有介绍。本来想装cuda8.0版本的,但是因为pytorch的cuda8.0没有torchvision所以选择了cuda9.0

        首先安装pytorch有很多方法。可以通过conda或者pip安装

conda安装虽然简单方便根据 官网 安装教程的一条命令安装,官网上也介绍了其他的安装方式,我也是参考着官方提供的方法进行安装的:

            

    conda install pytorch=0.4.1 cuda90 -c pytorch

这个访问的是国外的镜像源,基本是安装失败的。网上会有很多更换清华源的方法,但即使是添加了国内的源,也不一定能安装成功,首先要确定所添加的源里面是否包含你所需的包。所以这里直接跳过这种方法。

通过pip安装。

从网站上下载所需的包。如图所示:

ubuntu16.04 gtx1080ti 安装pytorch_第1张图片

ubuntu16.04 gtx1080ti 安装pytorch_第2张图片

 执行:

pip3 install torch-1.1.0-cp35-cp35m-linux_x86_64.whl
pip3 install torchvision-0.3.0-cp35-cp35m-manylinux1_x86_64.whl

 安装就完成了。

测试:当出现true的时候就pytorch cuda版本就安装成功了。

~/pytorch/cuda90$ python3.5
Python 3.5.2 (default, Jan 26 2021, 13:30:48)
[GCC 5.4.0 20160609] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> import torchvision
>>> torch.cuda.is_available()
True

总结:

        1.conda安装由于源的问题安装很慢,稍大一点的安装包几乎就是安装失败的。

        2.更换国内镜像源要保证所需要的的安装包是有的,不然也是安装失败的

        3.pip安装的时候要查看你的python版本,下载与你python版本对应的安装包,cp35对应python3.5, cp37对应python3.7,依次类推,否则也无法安装。

你可能感兴趣的:(pytorch,pytorch,深度学习,人工智能)