使用百度AI studio创建pytorch tensorflow gpu环境

最新推荐文章于 2024-05-25 20:43:36 发布

Bug永流传

最新推荐文章于 2024-05-25 20:43:36 发布

阅读量2.9k

点赞数

分类专栏：环境安装文章标签： tensorflow pytorch

本文链接：https://blog.csdn.net/qq_33590385/article/details/105904295

版权

环境安装专栏收录该内容

3 篇文章 0 订阅

订阅专栏

1. 配合视频服用更佳：

https://www.bilibili.com/video/BV16i4y1x7Yn/

2. 视频中使用到的命令：

mkdir /home/aistudio/cuda10
mkdir /home/aistudio/cuda92
mkdir /home/aistudio/cudnn
mkdir /home/aistudio/tf
cp /home/aistudio/data/data32924/cudnn-9.2.tgz /home/aistudio/cudnn
cp /home/aistudio/data/data32924/cudnn-10.1.tgz /home/aistudio/cudnn
pip install torch==1.5.0+cu101 torchvision==0.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html -t /home/aistudio/cuda10
pip install torch==1.5.0+cu92 torchvision==0.6.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html -t /home/aistudio/cuda92
pip install tensorflow-gpu==1.15 -t /home/aistudio/tf
tar -zxvf cudnn-9.2.tgz
tar -zxvf cudnn-10.1.tgz

import sys
sys.path.append('/home/aistudio/cuda92')
sys.path.append('/home/aistudio/tf')

import torch
import tensorflow as tf
print(torch.cuda.is_available())
print(tf.test.is_built_with_cuda())