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

墨风L

已于 2023-02-08 17:28:06 修改

阅读量475

点赞数

文章标签： pytorch 深度学习人工智能

于 2023-02-07 16:13:16 首次发布

本文链接：https://blog.csdn.net/lychangxiangsi/article/details/125064269

版权

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

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