王荣胜

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

2020-05-16 · 1 min read
百度

教程

命令

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

cd cudnn
tar -zxvf cudnn-9.2.tgz
tar -zxvf cudnn-10.1.tgz

进入GPU环境:

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())
小白
本文作者:王荣胜 |「邮箱 」| 「QQ」 | 「QQ群
文章出处:https://sqdxwz.top/
版权声明:本文章采用「知识共享署名-相同方式共享 4.0 国际许可协议」许可。
腾讯云新客专属福力
Powered by 王荣胜 | | 载入天数...载入时分秒...
Google     Server provider     jsDelivr    
正在加载今日诗词....