• NV驅(qū)動(dòng)下載安裝https://www.nvidia.cn/Download/index.aspx
• 多卡的話(huà)，安裝 NCCL https://developer.nvidia.com/nccl/nccl-download
• 安裝conda，并配置通道 vim ~/.condarc

show_channel_urls: truessl_verify: truechannels:- 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

• 創(chuàng)建虛擬環(huán)境

conda create -n torch_paddle python=3.8 conda activate torch_paddle

• 安裝 torch，注意都是用的 conda 安裝方式，pip 的安裝方式見(jiàn)文末參考鏈接

速度慢

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

這樣兩個(gè)框架就都被安裝在一個(gè)虛擬環(huán)境下了。

參考：
paddle安裝指導(dǎo)
torch安裝指導(dǎo)
TensorFlow 2.x GPU版在conda虛擬環(huán)境下安裝步驟
pip/conda導(dǎo)出 requirements.txt 注意事項(xiàng)
pip安裝pytorch1.10.1+paddlepaddle-gpu2.2.1+cuda10.2+cudnn7.6.5

總結(jié)

以上是生活随笔為你收集整理的conda安装pytorch1.10.1+paddlepaddle-gpu2.2.1+cuda10.2+cudnn7.6.5的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。