conda安装pytorch1.10.1+paddlepaddle-gpu2.2.1+cuda10.2+cudnn7.6.5

Michael阿明

发布于 2022-03-10 18:45:42

1.2K0

发布于 2022-03-10 18:45:42

NV驱动下载安装https://www.nvidia.cn/Download/index.aspx
多卡的话，安装 NCCL https://developer.nvidia.com/nccl/nccl-download
安装conda，并配置通道 vim ~/.condarc

show_channel_urls: true
  ssl_verify: true
  channels:
    - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
    - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
    - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
    - https://mirrors.bfsu.edu.cn/anaconda/cloud/peterjc123/
    - https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/
    - https://mirrors.bfsu.edu.cn/anaconda/cloud/menpo/
    - https://mirrors.bfsu.edu.cn/anaconda/cloud/bioconda/
    - https://mirrors.bfsu.edu.cn/anaconda/cloud/msys2/
    - https://mirrors.bfsu.edu.cn/anaconda/cloud/conda-forge/
    - https://mirrors.bfsu.edu.cn/anaconda/pkgs/main/
    - https://mirrors.bfsu.edu.cn/anaconda/pkgs/free/
    - https://mirrors.ustc.edu.cn/anaconda/cloud/menpo/
    - https://mirrors.ustc.edu.cn/anaconda/cloud/bioconda/
    - https://mirrors.ustc.edu.cn/anaconda/cloud/msys2/
    - https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
    - https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
    - https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
    - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
    - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
    - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
    - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
    - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
    - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
    - defaults

配置 pip 源 vim ~/.pip/pip.conf

[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple

创建虚拟环境

conda create -n torch_paddle python=3.8
conda activate torch_paddle

安装 torch，注意都是用的 conda 安装方式，pip 的安装方式见文末参考链接

速度慢

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

速度快

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

安装 paddle

conda install paddlepaddle-gpu==2.2.1 cudatoolkit=10.2 --channel https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/Paddle/

conda 包列表

# packages in environment at /anaconda3/envs/torch_paddle:
#
# Name                    Version                   Build  Channel
_libgcc_mutex             0.1                        main    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
_openmp_mutex             4.5                       1_gnu    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
astor                     0.8.1            py38h06a4308_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
blas                      1.0                         mkl    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
brotlipy                  0.7.0           py38h27cfd23_1003    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
bzip2                     1.0.8                h7b6447c_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
ca-certificates           2021.10.26           h06a4308_2    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
certifi                   2021.10.8        py38h06a4308_2    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
cffi                      1.14.6           py38h400218f_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
charset-normalizer        2.0.4              pyhd3eb1b0_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
cryptography              36.0.0           py38h9ce1e76_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
cudatoolkit               10.2.89              hfd86e86_1    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
cudnn                     7.6.5                cuda10.2_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
decorator                 5.1.0              pyhd3eb1b0_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
ffmpeg                    4.3                  hf484d3e_0    https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch
freetype                  2.11.0               h70c0345_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
gast                      0.3.3                      py_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
giflib                    5.2.1                h7b6447c_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
gmp                       6.2.1                h2531618_2    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
gnutls                    3.6.15               he1e5248_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
idna                      3.3                pyhd3eb1b0_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
intel-openmp              2021.4.0          h06a4308_3561    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
jpeg                      9d                   h7f8727e_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
lame                      3.100                h7b6447c_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
lcms2                     2.12                 h3be6417_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
ld_impl_linux-64          2.35.1               h7274673_9    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
libffi                    3.3                  he6710b0_2    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
libgcc-ng                 9.3.0               h5101ec6_17    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
libgomp                   9.3.0               h5101ec6_17    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
libiconv                  1.15                 h63c8f33_5    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
libidn2                   2.3.2                h7f8727e_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
libpng                    1.6.37               hbc83047_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
libprotobuf               3.19.1               h4ff587b_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
libstdcxx-ng              9.3.0               hd4cf53a_17    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
libtasn1                  4.16.0               h27cfd23_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
libtiff                   4.2.0                h85742a9_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
libunistring              0.9.10               h27cfd23_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
libuv                     1.40.0               h7b6447c_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
libwebp                   1.2.0                h89dd481_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
libwebp-base              1.2.0                h27cfd23_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
lz4-c                     1.9.3                h295c915_1    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
mkl                       2021.4.0           h06a4308_640    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
mkl-service               2.4.0            py38h7f8727e_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
mkl_fft                   1.3.1            py38hd3c417c_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
mkl_random                1.2.2            py38h51133e4_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
ncurses                   6.3                  h7f8727e_2    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
nettle                    3.7.3                hbbd107a_1    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
numpy                     1.21.2           py38h20f2e39_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
numpy-base                1.21.2           py38h79a1101_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
olefile                   0.46               pyhd3eb1b0_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
openh264                  2.1.1                h4ff587b_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
openssl                   1.1.1m               h7f8727e_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
paddlepaddle-gpu          2.2.1           py38_gpu_cuda10.2_many_linux    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/Paddle
pillow                    8.4.0            py38h5aabda8_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
pip                       21.2.4           py38h06a4308_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
protobuf                  3.19.1           py38h295c915_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
pycparser                 2.21               pyhd3eb1b0_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
pyopenssl                 21.0.0             pyhd3eb1b0_1    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
pysocks                   1.7.1            py38h06a4308_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
python                    3.8.12               h12debd9_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
pytorch                   1.10.1          py3.8_cuda10.2_cudnn7.6.5_0    https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch
pytorch-mutex             1.0                        cuda    https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch
readline                  8.1.2                h7f8727e_1    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
requests                  2.27.1             pyhd3eb1b0_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
setuptools                58.0.4           py38h06a4308_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
six                       1.16.0             pyhd3eb1b0_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
sqlite                    3.37.0               hc218d9a_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
tk                        8.6.11               h1ccaba5_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
torchaudio                0.10.1               py38_cu102    https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch
torchvision               0.11.2               py38_cu102    https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch
typing_extensions         3.10.0.2           pyh06a4308_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
urllib3                   1.26.7             pyhd3eb1b0_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
wheel                     0.37.1             pyhd3eb1b0_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
xz                        5.2.5                h7b6447c_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
zlib                      1.2.11               h7f8727e_4    https://mirrors.ustc.edu.cn/anaconda/pkgs/main
zstd                      1.4.9                haebb681_0    https://mirrors.ustc.edu.cn/anaconda/pkgs/main

这样两个框架就都被安装在一个虚拟环境下了。

参考：

paddle安装指导

torch安装指导

TensorFlow 2.x GPU版在conda虚拟环境下安装步骤

pip/conda导出 requirements.txt 注意事项

pip安装pytorch1.10.1+paddlepaddle-gpu2.2.1+cuda10.2+cudnn7.6.5

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

如有侵权请联系 cloudcommunity@tencent.com 删除

https