假设你要配置python==3.10,pytorch==2.1.0的环境
1.安装python 3.10
下载python 3.10源码
wget https://www.python.org/ftp/python/3.10.0/Python-3.10.0.tgz
解压源码
tar -xzf Python-3.10.0.tgz
cd Python-3.10.0
配置安装路径
./configure --prefix=$HOME/python3.10 --enable-optimizations
编译安装
make -j$(nproc)
make install
将python3.10 添加到path:
export PATH=$HOME/python3.10/bin:$PATH
验证安装:
python3.10 --version
2.创建虚拟环境
安装Python 3.10后,使用venv创建虚拟环境。
#创建虚拟环境
python3.10 -m venv myenv
#激活虚拟环境
source myenv/bin/activate
创建虚拟环境时,venv只是复制系统中已安装的Python解释器和标准库到虚拟环境目录中。 已经安装了Python 3.10,可以重复使用它来创建多个虚拟环境,而不需要重新下载。
3.安装pytorch
查看华为官网,选择要安装的版本:
安装torch_npu插件-安装步骤-配置与安装-配置与安装-开发文档-昇腾社区
将安装命令输入终端,即可安装:
安装结束!!