标签：torch 0gpu tensorflow2.0 pytorch1.4 正常 conda tensorflow

淡黄的炼丹炉（篇五）：conda+tensorflow2.0.0gpu+pytorch1.4.0gpu安装

• • 安装tensorflow2.0.0gpu
  • 测试tensorflow 是否正常
  • 安装pytorch1.4.0gpu
  • 测试pytorch 是否正常

安装tensorflow2.0.0gpu

conda create -n testenv python=3.6
conda activate testenv
conda install tensorflow-gpu==2.0.0

测试tensorflow 是否正常

import tensorflow as tf
print('GPU', tf.test.is_gpu_available())

安装pytorch1.4.0gpu

conda create -n torchenv python=3.6
conda activate torchenv
pip install torch==1.4.0+cu100 torchvision==0.5.0+cu100 -f https://download.pytorch.org/whl/torch_stable.html

测试pytorch 是否正常

import torch    # 如正常则静默
a = torch.Tensor([1.])    # 如正常则静默
a.cuda()    # 如正常则返回"tensor([ 1.], device='cuda:0')"
from torch.backends import cudnn # 如正常则静默
cudnn.is_acceptable(a.cuda())    # 如正常则返回 "True"

标签：torch,0gpu,tensorflow2.0,pytorch1.4,正常,conda,tensorflow
来源： https://blog.csdn.net/whstudio123/article/details/120610888

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。