GPU版的pytorch安装

2020-01-20  本文已影响0人  Bystander_1996

一:Anconda方式安装

1.1 安装anaconda

安装anaconda或者miniconda
进入清华镜像站下载自己需要的anaconda版本
anaconda下载地址

image.png
执行安装命令: bash + 下载的安装包
可能的出错是没有安装bzip2 安装一下sudo apt-get install bzip2
安装后如果找不到命令一般是由于环境变量未导入,或者环境变量导入了未更新。输入命令打开配置文件vim ~/.bashrc
在最后添加语句export PATH=/home/XXX/anaconda3/bin:$PATH
更新配置文件: source ~/.bashrc

1.2 anaconda更换清华镜像源

conda config --add channels [https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/](https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/)
conda config --set show_channel_urls yes       
//输入这两行命令即可换源成功。

anaconda常用命令

//更新所有库
conda update --all
//更新 conda 自身
conda update conda
//查看已经安装的包:
pip list 或者 conda list
//安装和更新:
pip install requests
pip install requests –upgrade
conda install requests
conda update requests

1.2 创建conda虚拟环境并进入:

conda create -n env_name pyhton=3.6  //创建环境 
source activate env_name  //激活并进入环境

1.3 安装pytorch

进入pytorch官网选择安装的命令 附:pytorch官网

conda安装方式
根据自己的系统,python语言版本等如实选择就可以得到安装命令,执行安装操作即可

二:pip安装

2.1 配置pip

更换pip的镜像站
修改 ~/.pip/pip.conf (没有就创建一个),修改内容如下

[global]
 index-url = https://pypi.tuna.tsinghua.edu.cn/simple

临时换源
在使用pip的时候加参数 -i https://pypi.tuna.tsinghua.edu.cn/simple
例:pip install gevent -i https://pypi.tuna.tsinghua.edu.cn/simple\

更新pip
python3 -m pip install --upgrade pip

2.2 安装pytorch

进入pytorch官网选择安装的命令 附:pytorch官网

pip安装方式

根据自己的系统,python语言版本等如实选择就可以得到安装命令,执行安装操作即可

三:查看是否安装成功

打开python

import torch
print(torch.cuda.is_available())
image.png
torch.cuda.is_available()的结果是True 证明是可以使用的,为了保险又用数据试了一下,也是没有问题的

参考:https://www.jianshu.com/p/e298b9d3afae

上一篇下一篇

猜你喜欢

热点阅读