1. 首頁 > 生活百科 > >

pytorch、paddlepaddle等環境搭建 深度學習環境搭建常用網址、conda/pip命令行整理( 二 )

生活百科


pytorch、paddlepaddle等環境搭建 深度學習環境搭建常用網址、conda/pip命令行整理

文章插圖
設置本機當前使用的cuda版本，只需要將PATH路徑中對應版本的bin文件目錄置前 。和CUDA_PATH、NVCUDASAMPLES_ROOT路徑無關 。
6. PyTorch環境搭建6.1 國內鏡像源使用（1）常見國內源鏡像
# 清華源pip install pkgname -i https://pypi.tuna.tsinghua.edu.cn/simple# 豆瓣源pip install pkgname -i https://pypi.douban.com/simple# 阿里源pip install pkgname -i https://mirrors.aliyun.com/pypi/simple# 百度源pip install pkgname -i https://mirror.baidu.com/pypi/simple# 中科大源pip install pkgname -i https://pypi.mirrors.ustc.edu.cn/simple（2）臨時使用源鏡像下載
# pippip install -i https://pypi.tuna.tsinghua.edu.cn/simple pkgnamepip install pkgname -i http://pypi.douban.com/simple/# condaconda install -c https://pypi.tuna.tsinghua.edu.cn/simple pkgnameconda install pkgname -c http://pypi.douban.com/simple/6.2 PyTorch安裝官網下載地址：Start Locally | PyTorch、Previous PyTorch Versions | PyTorch
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch （-c pytorch表示從官網下載）pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 –f http://download.pytorch.org/whl/torch_stable.html --trusted-host download.pytorch.org（速度快）使用官方網址提供的命令可快速完成torch安裝配置，但是官網只提供一些固定版本的torch和cuda組合 ?？梢愿鶕约旱挠布渲?，在官方下載網址或其他網站先行下載torch、torchvison等安裝包，然后將安裝包copy到指定路徑下，再用conda/pip命令安裝 。大多數安裝包都是以wheel格式保存的whl文件（Wheel是Python發行版的標準內置包格式），可直接用pip安裝 。
pip install cu101/torch-1.7.0%2Bcu101-cp38-cp38-win_amd64.whl6.3 測試torch環境import torch as tt.__version__# 成功則返回1.11.0+cu113t.cuda.is_available()# 成功則返回True# 退出python編譯環境：exit()# 或quit()7. PaddlePaddle環境搭建7.1 CPU版PaddlePaddleconda install paddlepaddle==2.3.2 --channel https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/Paddle/7.2 GPU版PaddlePaddle本人測試下來conda下載速度賊慢 ， pip很快，兩種方式都可嘗試下 ， 怎么快怎么來 。
conda install paddlepaddle-gpu==2.3.2 cudatoolkit=11.2 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/Paddle/ -c conda-forgepip install paddlepaddle-gpu==2.3.2.post112 -f https://www.paddlepaddle.org.cn/whl/windows/mkl/avx/stable.html7.3 測試PaddlePaddle環境import paddlepaddle.utils.run_check()電腦本地還沒有安裝cuda和cudnn，但也顯示GPU版本PaddlePaddle已可用，有點奇怪，后邊再訓練個網絡模型測試一下 。
pytorch、paddlepaddle等環境搭建 深度學習環境搭建常用網址、conda/pip命令行整理

文章插圖

推薦閱讀