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顺利安装