AI开发平台MODELARTS-GP Ant8裸金属服务器Ubuntu 20.04安装NVIDIA 470+CUDA 11.3:2、pytorch2.0安装和CUDA验证指南

时间：2024-09-05 08:36:23

AI开发平台MODELARTS

2、pytorch2.0安装和CUDA验证指南

PyTorch2.0所需环境为Python3.10，安装配置miniconda环境。

miniconda安装并创建alpha环境。

wget https://repo.anaconda.com/miniconda/Miniconda3-py310_23.1.0-1-Linux-x86_64.sh
chmod 750 Miniconda3-py310_23.1.0-1-Linux-x86_64.sh
bash Miniconda3-py310_23.1.0-1-Linux-x86_64.sh -b -p /home/miniconda
export PATH=/home/miniconda/bin:$PATH
conda create --quiet --yes -n alpha python=3.10

安装pytorch2.0并验证cuda状态。
在alpha环境下安装torch2.0，使用清华PIP源完成。
```
source activate alpha
pip install torch==2.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
python
```
验证torch与cuda的安装状态，输出为True即为正常。
```
import torch
print(torch.cuda.is_available())
```