标签:pytorch 1.3 安装 torch gpu Pytorch tensorflow Tensorflow CUDA10.0
介绍在CUDA 10.0的大环境下,安装pytorch-gpu(1.3.1)和tensorflow-gpu(1.15.0)
0. 环境
- CUDA 10.0
- cudnn 7.4
- python 3.7.2
- windows 10
1. 安装Pytorch-gpu
1.1 方法一: pytorch官网自动化安装命令
问题:仅支持CUDA10.2或者CUDA11.1,不放心
官网链接:https://pytorch.org/get-started/locally/#mac-installation
1.2 方法二:下载whl文件自己安装
下载地址:https://download.pytorch.org/whl/torch_stable.html
需要下载 torch 和 torchvision
我下载的版本是(一定选择cu开头的):
1.3 检验是否成功
import torch
print(torch.cuda.is_available())
print(torch.cuda.get_device_name(0))
2. 安装Tensorflow-gpu
2.1 找到对应版本直接安装
tensorflow_gpu与CUDA和cudnn版本对照表:
官网链接:https://tensorflow.google.cn/install/source#linux
我这里选择的是 tensorflow_gpu-1.15.0
pip install tensorflow_gpu==1.15.0
2.2 检验是否成功
import tensorflow as tf
print(tf.test.is_gpu_available())
标签:pytorch,1.3,安装,torch,gpu,Pytorch,tensorflow,Tensorflow,CUDA10.0 来源: https://www.cnblogs.com/bamlubi/p/15539720.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。