ubuntu18.04安装gpu版本的pytorch
一 安装anaconda1.下载Anaconda清华大学开源软件镜像站 https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/下载国内太慢镜像下载地址和修改方法参考博客:https://blog.csdn.net/wuzhiwuweisun/article/details/82689151下载版本Anaconda3-5.3.1-Linux-x8
一 安装anaconda
1.下载Anaconda
清华大学开源软件镜像站 Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror下载
国内太慢
镜像下载地址和修改方法参考博客:https://blog.csdn.net/wuzhiwuweisun/article/details/82689151
下载版本Anaconda3-5.3.1-Linux-x86_64.sh,自带python版本是3.7
2.安装Anaconda
首先切换到anaconda安装包的路径,在该路径下执行指令:
bash Anaconda3-5.0.1-Linux-x86_64.sh #后边的文件名称是你的安装包的名称
之后
Welcome to Anaconda3 5.0.1
In order to continue the installation process, please review the license
agreement.
Please, press ENTER to continue
>>> 按回车
#然后一直按回车到协议完毕
#出现:
Do you accept the license terms? [yes|no]
>>>输入yes
#下面就是问你安装目录,建议就是默认的安装路径,直接按回车
Anaconda3 will now be installed into this location:
/home/OO/anaconda3
- Press ENTER to confirm the location
- Press CTRL-C to abort the installation
- Or specify a different location below
[/home/mayunteng/anaconda3] >>> 按回车
选择其他目录 则[/home/mayunteng/anaconda3] >>> /media/mayunteng/Program/anaconda3 回车
#接下来就是等待安装完成
#注意安装完成以后会询问你是否把anaconda3的路径加到环境变量里去,一定要选yes,一定要选yes,一定要选yes。
询问是否安装visual studio code
报错Anaconda Extension Pack这个扩展包不可行,选择不安装
安装完成以后,重启终端,依次输入下面的指令,如果显示的是anaconda版本的python,代表安装成功。
先输入:source ~/.bashrc
再输入:python
显示的python版本后面有Anaconda标识,代表安装成功
Python 3.6.3 |Anaconda, Inc.| (default, Oct 13 2017, 12:02:49)
[GCC 7.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>>
#退出python的命令交互行输入exit()回车就可以了。
输入conda list 查看安装的库。
输入下面的指令可以启动Spyder, IPython, Jupyter Notebook,Anaconda Navigator 。
spyder #打开Spyder
ipython #打开IPython
jupyter notebook #打开Jupyter notebook
anaconda-navigator #打开Anaconda Navigator,打开anaconda-navigator时报错AttributeError: ‘str‘ object has no attribute ‘get‘,解决方法:进入"C:\Users\Administrator\Anaconda3\Lib\site-packages\anaconda_navigator\api\anaconda_api.py"查看原码line 903,修改version信息如下,这时vscode不再查检更新情况。
把
versions=[vsdata.get('productVersion')],
改成
versions=["1b8e8302e405050205e69b59abb3559592bb9e60"],
PS:若在前边没有把Anaconda3的路径加到环境变量中,可以这么修改:
在命令行输入:
sudo gedit ~/.bashrc
打开bashrc文件后在末尾输入下面的语句:
export PATH="/home/mayunteng/anaconda3/bin:$PATH" #/home/mayunteng/anaconda3/替换成你的安装路径
关闭bashrc文件后,在命令行输入:
source .bashrc #使更新的环境变量立即生效
输入以下指令,查看已有的环境变量,确认是否添加了anaconda的安装路径
echo $PATH
二 系统检查
确认系统可以使用cuda,并且确认与系统GPU兼容的CUDA/ cuDNN包,如图所示系统的driver version时465.27,当前驱动支持的最高的cuda版本是11.3,cuda是向下兼容的
三 安装gpu版本的pytorch
1.创建一个虚拟环境
conda create -n torch-gpu python=3.7
source activate torch-gpu
可能的报错
conda: command not found
This occurs because the path for anaconda installation has not been set in your .bashrc
or .zshrc
Try:
export PATH="/home/username/anaconda3/bin:$PATH"
2.安装pytorch
For PyTorch it is straight forward than TensorFlow installation because you don’t have to separately install CUDA ToolKit and cuDNN because you can install them at once using a single command as given in PyTorch official website.
进入https://pytorch.org/get-started/locally/网址,it automatically select the appropriate pytorch version depending on your operating system, its version and CUDA version.
上图给出的是适合本系统的pytorch版本、cuda版本,以及安装指令
激活创建的虚拟环境,在该环境中安装pytorch
四 验证是否安装成功
但是激活当前环境,启动spyder验证时报错ModuleNotFoundError: No module named 'torch'
这是因为Anaconda自带的spyder使用的是base环境,conda上配置好的虚拟环境并不能直接搭载到spyder上,需要在anaconda中按照图片所示在当前虚拟环境下安装spyder 与spyder -kernels
可以直接在激活虚拟环境下直接运行pip install spyder或者conda install spyder
Ubuntu18.04下安装深度学习框架Pytorch(GPU加速)_wuzhiwuweisun的博客-CSDN博客_ubuntu18.04安装pytorch
其他安装过程可参考在Ubuntu18.04上搭建Pytorch深度学习环境 - 知乎
更多推荐
所有评论(0)