Ubuntu安装Pytorch方法

Ubuntu安装Pythorch需要先屏蔽掉系统自带的显卡驱动器,安装合适的NVIDIA显卡驱动,方法参考显卡安装篇;
ubuntu电脑需要安装anaconda,新建TORCH环境。

我的电脑老旧,显卡版本太低,无法安装最新版本pytorch-1.4.0,只能安装pytorch-1.2.0,才能运行cuda运算加速。

step1.添加国内conda下载镜像地址:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

step2.安装pytorch与torchvision

conda install pytorch==1.2.0 torchvision

我的电脑用conda安装完毕还是无法调用torch,还要再用pip安装一下:

pip --default-timeout=100000 install torch==1.2.0 -i https://mirrors.aliyun.com/pypi/simple/

step3.检测pytorch是否安装完成
进入python

>>> import torch
>>> torch.cuda.is__available()
>>> torch.ones(5,4).cuda()

三组命令都无问题,才说明pytorch顺利安装

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容