anaconda PyTorch 環境架設(ubuntu 20.04)

• anaconda PyTorch 環境架設(ubuntu 20.04)
  • 1. Anaconda 下載與安裝
  • 2. 使用 Anaconda Prompt(anaconda3) 創建 python3.7 環境
  • 3.安裝 PyTorch
  • 4.安裝 conda 相關套件

1. Anaconda 下載與安裝

Anaconda下載

2. 使用 Anaconda Prompt(anaconda3) 創建 python3.7 環境

創造一個 conda for PyTorch 環境

conda create -n pytorch python=3.7

切進環境

conda activate pytorch

3.安裝 PyTorch

獲得安裝指令

conda activate pytorch
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

安裝後驗證

conda activate pytorch
python
import torch

import torchvision
torchvision.__version__

驗證 CUDA 是否可用

torch.cuda.is_available()
torch.cuda.get_device_name()

torch.cuda.device_count()

CUDA ID

torch.cuda.current_device()

CUDA 裝置名稱

torch.cuda.get_device_name(0)

4.安裝 conda 相關套件

pip install jupyterlab
conda install -c conda-forge matplotlib -n pytorch
conda install -c anaconda pandas -n pytorch
pip install torchsummary
pip install torch
pip install opencv-python
pip install opencv-contrib-python

tags: setup, Python