Ubuntu Python机器学习框架如何搭建
在Ubuntu上搭建Python机器学习框架可按以下步骤进行:
-
安装基础工具
更新系统并安装Python和pip:sudo apt update sudo apt install python3 python3-pip
验证安装:
python3 --version pip3 --version
-
创建虚拟环境(可选但推荐)
隔离项目依赖:sudo apt install python3-venv python3 -m venv myenv source myenv/bin/activate
-
安装机器学习库
- 基础库:NumPy、Pandas、Scikit-learn
pip install numpy pandas scikit-learn
- 深度学习框架:
- TensorFlow:
pip install tensorflow # 如需GPU支持,安装对应版本并配置CUDA/cuDNN
- PyTorch:
pip install torch torchvision torchaudio # 如需GPU支持,安装对应版本并配置CUDA
- TensorFlow:
- 其他工具:Matplotlib(可视化)、OpenCV(计算机视觉)
pip install matplotlib opencv-python
- 基础库:NumPy、Pandas、Scikit-learn
-
验证环境
运行简单代码测试库是否正常:- Scikit-learn示例(线性回归):
import numpy as np from sklearn.linear_model import LinearRegression X = np.random.rand(100, 1) y = 2 + 3 * X model = LinearRegression().fit(X, y) print(model.predict([[0.5]]))
- TensorFlow示例(MNIST分类):
import tensorflow as tf from tensorflow.keras import layers (x_train, y_train), _ = tf.keras.datasets.mnist.load_data() x_train = x_train / 255.0 model = tf.keras.Sequential([layers.Flatten(), layers.Dense(10, activation='softmax')]) model.compile(optimizer='adam', loss='sparse_categorical_crossentropy') model.fit(x_train, y_train, epochs=1)
- Scikit-learn示例(线性回归):
-
开发工具(可选)
- Jupyter Notebook:
pip install jupyter jupyter notebook
- VS Code:安装Python扩展,直接运行脚本。
- Jupyter Notebook:
注意事项:
- 深度学习框架需GPU支持时,需提前安装CUDA和cuDNN,并确保版本兼容。
- 可通过
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
切换国内镜像源加速安装。
以上步骤参考自,可根据具体需求调整库的安装组合。
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权请联系我们,一经查实立即删除!