本文主要是介绍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了吗?
这篇关于Aistudio使用经验积累(持续更新中)的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!