快速安装torch-gpu和Tensorflow-gpu(自用,Ubuntu)

要更详细的教程可以参考Tensorflow + PyTorch 安装(CPU + GPU 版本),这里是有基础之后的快速安装。

一、Pytorch

  • 安装
bash 复制代码
conda create -n torch_env python=3.10.13
conda activate torch_env
conda install cudatoolkit==11.8 -c nvidia
pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118
conda install -c "nvidia/label/cuda-11.8.0" cuda-nvcc
  • 测试
python 复制代码
# 进入python
python
import torch
print(torch.cuda.is_available())

二、Tensorflow

  • 安装

安装完Pytorch,再安装Tensorflow

bash 复制代码
conda create -n tensorflow_env python=3.7
conda activate tensorflow_env
conda install cudatoolkit==11.8 -c nvidia
pip install tensorflow-gpu==2.6.0

# 测试时会报错说LD_LIBRARY_PATH: :/usr/local/cuda/lib64下没有libcudnn.so.8
# 于是用locate在电脑中查找
# 显示其中一个路径是 your_path/Anaconda3/envs/torch_env/lib/python3.10/site-packages/torch/lib/libcudnn.so.8,复制到/usr/local/cuda/lib64
sudo updatedb
locate libcudnn.so.8   
sudo cp your_path/Anaconda3/envs/torch_env/lib/python3.10/site-packages/torch/lib/libcudnn.so.8 /usr/local/cuda/lib64
  • 测试
python 复制代码
import tensorflow as tf
print(tf.test.is_gpu_available())

三、查看显卡利用率

bash 复制代码
# 简单查看
nvidia-smi
# 每2秒刷新
nvidia-smi -l 1
相关推荐
Piink6 分钟前
网络模型训练完整代码
人工智能·深度学习·机器学习
曾经的三心草6 分钟前
OpenCV4-直方图与傅里叶变换-项目实战-信用卡数字识别
python·opencv·计算机视觉
luoganttcc9 分钟前
在 orin 上 安装了 miniconda 如何使用 orin 内置的 opencv
人工智能·opencv·计算机视觉
JinchuanMaster11 分钟前
cv_bridge和openCV不兼容问题
人工智能·opencv·计算机视觉
Rhys..20 分钟前
python自动化中(包括UI自动化和API自动化)env的作用和使用
python·ui·自动化
心勤则明25 分钟前
Spring AI 文档ETL实战:集成text-embedding-v4 与 Milvus
人工智能·spring·etl
啦啦啦在冲冲冲26 分钟前
mse和交叉熵loss,为什么分类问题不用 mse
人工智能·分类·数据挖掘
SaaS_Product28 分钟前
有安全好用且稳定的共享网盘吗?
人工智能·云计算·saas·onedrive
~~李木子~~30 分钟前
图像分类项目:Fashion-MNIST 分类(SimpleCNN )
人工智能·分类·数据挖掘
轻赚时代34 分钟前
新手做国风视频难?AI + 敦煌美学高效出片教程
人工智能·经验分享·笔记·创业创新·课程设计·学习方法