离线安装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
发布评论