快速安装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
相关推荐
湫ccc7 分钟前
《Python基础》之pip换国内镜像源
开发语言·python·pip
hakesashou8 分钟前
Python中常用的函数介绍
java·网络·python
菜鸟的人工智能之路18 分钟前
极坐标气泡图:医学数据分析的可视化新视角
python·数据分析·健康医疗
菜鸟学Python19 分钟前
Python 数据分析核心库大全!
开发语言·python·数据挖掘·数据分析
小白不太白95020 分钟前
设计模式之 责任链模式
python·设计模式·责任链模式
WeeJot嵌入式24 分钟前
卷积神经网络:深度学习中的图像识别利器
人工智能
喜欢猪猪26 分钟前
Django:从入门到精通
后端·python·django
糖豆豆今天也要努力鸭32 分钟前
torch.__version__的torch版本和conda list的torch版本不一致
linux·pytorch·python·深度学习·conda·torch
脆皮泡泡33 分钟前
Ultiverse 和web3新玩法?AI和GameFi的结合是怎样
人工智能·web3
机器人虎哥37 分钟前
【8210A-TX2】Ubuntu18.04 + ROS_ Melodic + TM-16多线激光 雷达评测
人工智能·机器学习