pytoch 安装为Gpu版本

1，首先确保自己有显卡，然后根据显卡安装cuda ,nvidia

2,cmd 打开终端

查看自己的python版本

在下面的链接中找到相应版本的torch.torchvision

https://download.pytorch.org/whl/torch_stable.html

一定要确保自己的torch与torchvision对应

下面是一个对应表

Gpu版本是cu开头的，

然后直接pip install 文件名.whl即可，如果出现以下这种情况，在最后面加入--user重新安装（因为权限不够）

最后安装成功！

测试一下是否安装成功

在cmd终端输入以下代码段

python
import torch
torch.cuda.is_available()

显示True 则安装成功

pytoch 安装为Gpu版本相关推荐

windows10下安装tensorflow(gpu版本)
windows10安装tensorflow的gpu版本(pip3安装方式) 前言: TensorFlow 有cpu和 gpu两个版本:gpu版本需要英伟达CUDA 和 cuDNN 的支持,cpu版本不 ...
Ubuntu 20.04 安装Tensorflow GPU版本（GTX-1060显卡）
Ubuntu 20.04 安装Tensorflow GPU版本 (GTX-1060显卡) 参考: https://cyfeng.science/2020/05/02/ubuntu-install-nv ...
通过Anaconda在Ubuntu16.04上安装 TensorFlow(GPU版本)
一. 安装环境 Ubuntu16.04.3 LST GPU: GeForce GTX1070 Python: 3.5 CUDA Toolkit 8.0 GA1 (Sept 2016) cuDNN v6 ...
ubuntu 18 安装xgboost GPU版本
先把之前安装的xgboost都卸载干净: pip uninstall xgboost 确保电脑里安装了cuda. 然后按照下面的步骤来安装GPU版本的xgboost: git clone --recu ...
win7 64位安装tensorflow GPU版本
近日终于在win7上安装成功tensorflow GPU版本,并测试成功!这里将其中遇到的种种问题和解决方法做下记录,以供大家相互交流! 电脑配置:16G内存:处理器: AMD Athlon(tm ...
WINDOWS 安装XGBoost GPU版本最新简易方法
目录一.系统配置二.问题背景三.执行步骤 1. 安装cuda a. 检查是否安装了CUDA b. 从dos中查看可以支持的cuda版本 c 下载对应版本的cuda d 根据引导安装cuda 2. ...
AArch64编译安装特定GPU版本TensorFlow及Bazel
前排提示:如果使用的 cuDNN 版本高于 7,会无法编译安装带 CUDA 支持的 TensorFlow 1 版本. 本文基于 Jetson TX2. ALBERT 的 requirements.tx ...
安装pytorch gpu版本
在windows下用anaconda虚拟环境安装pytorch gpu版成功! 文章目录 0. 用pip安装 1. 安装准备 2. 安装指令 3. 查看.提升cuda版本 3.1 判断显卡是否支持c ...
TensorFlow学习（五）之Ubuntu16.04安装TensorFlow———GPU版本（gtx1070+ubuntu16.04）
Ubuntu16.04系统安装系列: Ubuntu系统安装QQ+百度云盘+微信+迅雷下载等 Ubuntu系统安装网易云音乐 Ubuntu系统安装搜狗输入法 Ubuntu系统安装SMPlayer播放器 ...

pytoch 安装为Gpu版本

pytoch 安装为Gpu版本相关推荐

最新文章

热门文章