Pytorch利用Anaconda搭建虛擬環境+cuda+cudnn


1.下載電腦NVIDIA的顯卡驅動

2. 下載安裝相應的CUDA +CUDNN

3. 安裝pytorch

  • 3.1 利用Anaconda 創建虛擬環境

    • 打開Anaconda Prompt,在命令行輸入命令 :
      conda create -n nanme python=3.8 創建虛擬環境
      conda env list 查看創建的虛擬環境
      conda activate name 激活虛擬環境

    • 添加鏡像源:
      conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
      conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
      conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
      conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
      conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
      conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
      conda config --set show_channel_urls yes
      conda config --show channels

  • 3.2 torch torchvision cuda python對應版本

    • torch torchvision cuda python需要對應才可以正常工作
  • 3.3 安裝

    • 3.3.1 激活虛擬環境

      • 在打開Anaconda Prompt,輸入 conda activate name
        • name 為所創建的虛擬環境名字
    • 3.3.2 安裝pytorch、torchvision

    • 3.3.3 測試是否安裝成功
      create activate name 激活虛擬環境
      python' 進入python模式 import torch 導入torchtorch.cuda.is_available()` 查看cuda是否能能使用,若可以使用返回True


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM