Ubuntu18.04离线安装pytorch

编程入门 行业动态 更新时间:2024-10-24 22:24:38

Ubuntu18.04<a href=https://www.elefans.com/category/jswz/34/1767604.html style=离线安装pytorch"/>

Ubuntu18.04离线安装pytorch

由于没有网络,只能离线安装,下载对应的包,进行安装

1.安装显卡驱动:根据购买的显卡型号进行安装

下载地址:.aspx?lang=cn
安装教程可以参考:.aspx?lang=cn
我是购买的时候厂家就给我安装好了,所以我就直接安装cuda+cudnn了。
如果是自己安装显卡驱动的话,注意了,需要切换到F2的命令行去安装,不能ubuntu的桌面进行安装,我的是两张3080Ti,驱动是cuda11.3

2.安装CUDA+CUDNN

cuda的历史版本:
cudnn的历史版本:
我下载的是CUDA10.1, 最后换成了11.3; cudnn与cuda一定要版本对应,安装过程可以参考:

sudo sh cuda_10.1.243_418.87.00_linux.run
我在这里之前下载错版本了,就安装失败,所以一定要看看下载下来了是不是版本的问题

输入accept进行后续操作,随即出现以下界面:

因为我们在安装cuda之前已经安装了Nvidia的驱动,因此这里的第一项我们必须取消勾选,选择不安装驱动,随后选择Install进行后续操作。

打开命令行输入以下命令进行~/.bashrc文件的修改:
我是在root下的/.bashrc与用户下的/.bashrc都修改了

vim ~/.bashrc

在最后添加:

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda

我是看的别人这样添加的,但是我是安装了pytorch之后缺少动态库:
问题是这样的:

ImportError: libnvToolsExt.so.1: cannot open shared object file: No such file or directory

参考了:
所以就修改为了:

export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.1

千万不要忘记了source ~/.bashrc

nvcc移动至/usr/bin/

sudo cp /usr/local/cuda-10.1/bin/nvcc /usr/bin/

验证cuda是否成功

nvcc --version

安装cudnn

sudo cp cuda/include/cudnn*.h /usr/local/cuda/include     
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64  
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* 

3.安装ananconda3

参考:

4.安装pytorch

下载对应的torch1.3版本,注意显卡的版本我是对应了之前的cuda10.1
/

本地安装可以:

conda install -n env_name --use-local pytorch-1.3.0-py3.7_cuda10.0.130_cudnn7.6.3_0.tar

注意:这里选择的是具有cudnn的torch版本

注意:其他版本opencv3.4 , numpy1.18
opencv4.5安装失败,numpy1.19安装失败。那么高的版本好像不兼容

5.pycharm中run **.py还是出现了libnvToolsExt.so.1问题

ImportError: libnvToolsExt.so.1: cannot open shared object file: No such file or directory
参考:

sudo ln -s /usr/local/cuda-10.0/lib64/libnvToolsExt.so.1  /home/用户名/bin
sudo ldconfig

其中:/home/用户名/bin 添加了环境变量

我的案例:

sudo ln -s /usr/local/cuda-10.0/lib64/libnvToolsExt.so.1  /home/hj/bin
sudo ldconfig

其中地址:/home/hj/bin 被添加到~/.bashrc环境变量里面去了

更多推荐

Ubuntu18.04离线安装pytorch

本文发布于:2024-02-26 13:20:15,感谢您对本站的认可!
本文链接:https://www.elefans.com/category/jswz/34/1702624.html
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。
本文标签:离线   pytorch

发布评论

评论列表 (有 0 条评论)
草根站长

>www.elefans.com

编程频道|电子爱好者 - 技术资讯及电子产品介绍!