pytorch-gpu安装（anaconda3+cuda10.2.89+pytorch1.7）(仅适用于30系以下的显卡)

Posted 2023-03-01 墨风L

（自己学习使用，有什么不对的感谢指出。）

1、安装anaconda

官网下载即可，注意与python版本对应问题。（本文中是anaconda3-5.3.1-Windows-x86_64）使用conda --version测试是否安装成功。

anaconda：https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
或：https://docs.anaconda.com/anaconda/packages/oldpkglists/
miniconda：https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/?C=N&O=D

环境变量设置：

2、安装cuda

cuda安装包（10.2.89）
(https://developer.nvidia.com/cuda-10.2-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=deblocal)（CUDA Toolkit 10.2 Download | NVIDIA Developer）

win下安装cuda（先建立环境并激活）
安装cuda：conda install cudatoolkit=10.2.89
安装cudnn：conda install cudnn=7.6.5

1.若已经安装过CUDA，先删除掉已安装的CUDA历史版本的整个文件夹
一般默认的CUDA目录位于：C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\
2.安装CUDA 10.2时不要勾选CUDA中的Visual Studio Intergration

3、安装pytorch

        a、打开anaconda prompt，创建新环境
                conda create --name 文件名 python==版本号

        b、安装pytorch

                PyTorch安装命令 ：https://pytorch.org/get-started/locally/

                conda install pytorch=1.7.0 torchvision torchaudio cudatoolkit=10.2 -c pytorch
                Ubuntu中可以这样执行，直接会有cuda10.2.89

        有时可以用以下这几个网址：

      

4、安装opencv-python的whl包    

        下载对应的whl版本：https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple/opencv-python/

        python3.7 下载的是opencv_python-4.1.0.25-cp37-cp37m-manylinux1_x86_64.whl

        进入安装包所在目录
        例如：Ubuntu：cd /home/ubuntu/user_space/torch1.7.1-cuda10.1
                   Win10：cd /d E:\\anaconda/envs/pytorch37

        pip install opencv_python-4.1.0.25-cp37-cp37m-manylinux1_x86_64.whl

5、有些没有安装较新的驱动

        需要在NVIDIA GeForce 驱动程序 - N 卡驱动 | NVIDIA里面安装驱动。

6、常用操作

        激活环境activate name
        退出环境conda deactivate
        复制环境conda create -n name2 --clone name1（不能直接重命名）
        卸载环境conda uninstall -n name --all
        删除环境conda remove -n name --all
        安装指定源的包

pip --default-timeout=100 install packagename -i https://pypi.douban.com/simple
其他一些源： pip install -i https://pypi.douban.com/simple
            pip install -i http://pypi.hustunique.com/simple 
            http://pypi.sdutlinux.org/simple 
            http://pypi.mirrors.ustc.edu.cn/simple 
清华源：https://pypi.tuna.tsinghua.edu.cn/simple