Linux下的pytorch安装及踩坑记录（亲测有效2022.11.23）
Linux下的pytorch安装及踩坑记录（亲测有效2022.11.23）
1、安装环境介绍：
1.1、显卡配置：RTX3060 桌面系统：ubuntu18.04
1.2、根据1.1的配置选择对应的Python、CUDA、pytorch、torchvision,我选择Python = 3.8| CUDA 11.7 |pytorch =1.13 | torchvision= 0.11
注意：这几个安装包的版本需要对应起来，版本不对会产生错误！！！
2、创建anaconda虚拟环境及conda换源：
2.1、安装anaconda和创建anaconda虚拟环境
网上教程很多，可以直接去官网安装即可。
2.2、conda换源
要更换清华或者中科大的源，后期下载的速度才不会i被限制：
命令：查看主目录下所有文件
ls -a
编辑.condarc
gedit ./.conda
将下面的源拷贝到文件中并保存。
channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
show_channel_urls: true
auto_activate_base: false
3、pytorch、torchvision下载安装：
3.1、确认安装前的准备工作是否完成
激活conda工作环境、查看python的版本、查看安装的CUDA的版本，命令如下：
meng@meng-PC:~$ conda activate pytorch_env
(pytorch_env) meng@meng-PC:~$ python --version
Python 3.8.0
(pytorch_env) meng@meng-PC:~$ nvidia-smi
结果如图：
3.2、pytorch、torchvision安装
方法一：通过pytorch官网选择安装，因为速度太慢，本文没有选择这种方式
方法二：去清华源直接下载需要的包，直接安装
1、清华源镜像网站寻找自己所需要的版本，注意分清楚GPU和CPU版本
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/
2、安装pytorch、torchvision
cd到自己的安装目录下运行下面指令，完成pytorch的安装
conda install pytorch-1.13.0-py3.8_cuda11.7_cudnn8.5.0_0.tar.bz2
同样的
conda intalll torchvision-0.11.0-py38_cu113.tar.bz2
完成torchvision的安装
3、测试是否安装成功
(pytorch_env) meng@meng-PC:~$ python
Python 3.8.0 (default, Nov 6 2019, 21:49:08)
[GCC 7.3.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
True
>>>
如果结果为ture恭喜你torch的环境安装成功，但是实际情况往往不是这样，下面是我的踩坑记录：
3.3、踩坑记录
1、报错信息如下：
(pytorch_env) meng@meng-PC:~/下载$ python
Python 3.8.0 (default, Nov 6 2019, 21:49:08)
[GCC 7.3.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
Traceback (most recent call last):
File "" , line 1, in <module>
File "/home/meng/anaconda3/envs/pytorch_env/lib/python3.8/site-packages/torch/__init__.py", line 191, in <module>
_load_global_deps()
File "/home/meng/anaconda3/envs/pytorch_env/lib/python3.8/site-packages/torch/__init__.py", line 153, in _load_global_deps
ctypes.CDLL(lib_path, mode=ctypes.RTLD_GLOBAL)
File "/home/meng/anaconda3/envs/pytorch_env/lib/python3.8/ctypes/__init__.py", line 369, in __init__
self._handle = _dlopen(self._name, mode)
OSError: libmkl_intel_lp64.so: cannot open shared object file: No such file or directory
解决：conda的环境变量不在当前用户的搜索路径中，先打开当前用户环境变量配置文件bashrc，将
export LD_LIBRARY_PATH=/home/jc/anaconda3/lib:$LD_LIBRARY_PATH
其中jc为用户名
参考文章：https://blog.csdn.net/Christine_11/article/details/126943635
2、报错信息二
>>> import torch
Traceback (most recent call last):
File "" , line 1, in <module>
File "/home/meng/anaconda3/envs/pytorch_env/lib/python3.8/site-packages/torch/__init__.py", line 753, in <module>
from .serialization import save, load
File "/home/meng/anaconda3/envs/pytorch_env/lib/python3.8/site-packages/torch/serialization.py", line 18, in <module>
from typing_extensions import TypeAlias
ModuleNotFoundError: No module named 'typing_extensions'
这个简单，直接安装typing_extensions即可
指令;
pip install typing_extensions
3.4、本文安装包下载路径
https://mp.csdn.net/mp_download/manage/download/UpDetailed