其他分享
首页 > 其他分享> > Aistudio使用经验积累(持续更新中)

Aistudio使用经验积累(持续更新中)

作者:互联网

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())


nvcc --version#查看cuda版本


问题:
aistudio是不能运行tf了吗?

标签:积累,tgz,Aistudio,torch,更新,cudnn,tf,home,aistudio
来源: https://blog.csdn.net/a18829292719/article/details/117948222