GPU版的pytorch安装
2020-01-20 本文已影响0人
Bystander_1996
一:Anconda方式安装
1.1 安装anaconda
安装anaconda或者miniconda
进入清华镜像站下载自己需要的anaconda版本
anaconda下载地址
执行安装命令:
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官网
根据自己的系统,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官网
根据自己的系统,python语言版本等如实选择就可以得到安装命令,执行安装操作即可
三:查看是否安装成功
打开python
import torch
print(torch.cuda.is_available())
image.png
torch.cuda.is_available()的结果是True 证明是可以使用的,为了保险又用数据试了一下,也是没有问题的