尝试在pyTorch中使用Cuda
在pyTorch中使用CUDA是为了利用GPU加速深度学习模型的训练和推理过程。CUDA是NVIDIA提供的并行计算平台和编程模型，可以利用GPU的并行计算能力加速计算任务。
在pyTorch中使用CUDA需要进行以下步骤：
- 检查CUDA是否可用：首先需要检查系统中是否安装了NVIDIA的显卡驱动和CUDA工具包。可以通过运行
nvidia-smi命令来查看显卡驱动的版本和CUDA的安装情况。 - 安装pyTorch和CUDA支持：确保已经安装了适用于CUDA的pyTorch版本。可以通过官方网站或者pip安装命令来获取相应的版本。
- 将模型和数据移动到GPU：在使用CUDA加速之前，需要将模型和数据移动到GPU上。可以使用
model.to('cuda')将模型移动到GPU上，使用input.to('cuda')将输入数据移动到GPU上。 - 设置优化器和损失函数：在使用CUDA加速时，需要将优化器和损失函数与GPU相关联。可以使用
optimizer = torch.optim.SGD(model.parameters(), lr=0.1).to('cuda')将优化器与GPU相关联。 - 执行模型训练和推理：在使用CUDA加速后，可以像正常情况下一样执行模型的训练和推理过程。pyTorch会自动利用GPU进行计算。
尽管pyTorch提供了CUDA加速的功能，但在使用过程中需要注意以下几点：
- GPU内存限制：显卡的内存是有限的，如果模型或数据过大，可能会导致内存不足的问题。可以通过减小批量大小、使用更小的模型或者使用分布式训练等方式来解决。
- 数据类型匹配：在使用CUDA加速时，需要确保模型和数据的数据类型匹配。通常情况下，模型和数据都应该使用
torch.cuda.FloatTensor类型。 - GPU加速效果：并不是所有的计算任务都适合使用GPU加速，有些任务可能在CPU上执行更快。因此，在使用CUDA加速之前，需要评估任务的特点和需求，确定是否适合使用GPU加速。
推荐的腾讯云相关产品和产品介绍链接地址：
- 腾讯云GPU云服务器：提供了强大的GPU计算能力，适用于深度学习、科学计算等任务。详情请参考：https://cloud.tencent.com/product/cvm/gpu
- 腾讯云AI引擎：提供了基于GPU的深度学习推理服务，支持常见的深度学习框架和模型。详情请参考：https://cloud.tencent.com/product/tia
- 腾讯云容器服务：提供了基于Kubernetes的容器管理服务，可以方便地部署和管理深度学习模型。详情请参考：https://cloud.tencent.com/product/tke
相关·内容
3回答
我今天安装了蟒蛇、CUDA和PyTorch，但我无法在torch中访问我的GPU (RTX2070)。我遵循了所有的安装步骤，PyTorch在其他方面都运行得很好，但是当我尝试在shell或脚本中访问GPU时，我得到
>>> import torch
>>> torch.cuda.is_available()
False
>>> torch.cuda.device_count()
0
>>> print(torch.version.cuda)
None
cudatoolkit 11.3.1
浏览 7提问于2021-10-24得票数 0
1回答
一开始，我使用nvcc --version命令检查了我的cuda版本，它将版本显示为10.2，因此我开始根据中的指令安装，所以我尝试在anaconda提示符中使用bellow命令使用python 3.8.3虚拟环境
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
在终端显示
added / updated specs:
- cudatoolkit=10.2
- pytorch
- torchaudio
- torchvisio
浏览 6提问于2020-11-16得票数 1
我已经通过运行以下命令安装了pytorch cuda：
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia
我的cuda版本是11.2。我正在使用windows 10。Pytorch cuda 11.2目前不可用。()
所以我已经安装了11.1版本。
(使用nvidia-smi)
但它显示为假。
torch.cuda.is_available() >>> False
我尝试过10.2和11.1两个版本。
据我所知，我不需要安装pytorch的cuda工具包。
浏览 85提问于2021-05-17得票数 1
PyTorch未启用和访问GPU驱动程序和CUDA。torch.cuda.is_available()返回false
我使用的是macOS Mojave 10.14.6，我已经安装了cuda10.0版本的pytorch.我在上尝试了验证，构造一个随机初始化的张量工作得很好。
但当我试着
import torch
torch.cuda.is_availalbe()
它返回false。
因此，我按照Pytorch上的说明安装了Anaconda和Cuda。
然后尝试这样做：
git clone --recursive https://github.com/pytorch/pytorch cd pyt
浏览 8提问于2019-09-16得票数 2
我是PyTorch新手，错误地安装了CUDA10.2版的PyTorch。实际上，我的系统中没有CUDA。因此，当我编译我的程序时，我得到以下错误：
RuntimeError: Detected that PyTorch and torch_cluster were compiled with different CUDA versions. PyTorch has CUDA version 10.2 and torch_cluster has CUDA version 0.0. Please reinstall the torch_cluster that matches your PyTor
浏览 31提问于2021-01-24得票数 0
当我试图将PyTorch从1.4.0修改为1.5.0时，Anaconda说所有的软件包都已经安装好了。
$ conda install -c pytorch pytorch torchvision
Collecting package metadata (current_repodata.json): done
Solving environment: done
# All requested packages already installed.
$ conda list | grep -i torch
_pytorch_select 0.2
浏览 2提问于2020-04-24得票数 5
回答已采纳
我正在尝试用pip安装torch_scatter。然而，它给了我一个错误消息：
File "/home1/huangjiawei/miniconda3/envs/lin/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 404, in build_extensions
self._check_cuda_version()
File "/home1/huangjiawei/miniconda3/envs/lin/lib/python3.8/site-pac
浏览 25提问于2022-03-16得票数 1
我试图在A100图形处理器上使用我当前的代码，但是我得到了这个错误：
---> backend='nccl'
/home/miranda9/miniconda3/envs/metalearningpy1.7.1c10.2/lib/python3.8/site-packages/torch/cuda/__init__.py:104: UserWarning:
A100-SXM4-40GB with CUDA capability sm_80 is not compatible with the current PyTorch installation.
The curre
浏览 12提问于2021-04-08得票数 9
我安装了几个PyTorch版本(CUDA 11.7 nightly、CUDA 11.6 nightly、11.3)，但每次torch.version.cuda都返回10.2。
我想在CUDA 11.7上运行PyTorch。我的显卡有CUDA功能sm_86。
[me@legion imagen-test]$ sudo pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
...
[me@legion imagen-test]$ python
>&
浏览 30提问于2022-05-29得票数 2
回答已采纳
我如何通过终端或导航器使用Anaconda从1.4 -> 1.5更新PyTorch？
使用conda update --all更新蟒蛇更新了一些包，而不是所有包，包括PyTorch。
最初，我通过运行conda install -c pytorch pytorch安装PyTorch
在PyTorch Github页面上有以下命令
conda install -c pytorch magma-cuda90 # or [magma-cuda92 | magma-cuda100 | magma-cuda101 ] depending on your cuda version
但我想知道已经安装
浏览 0提问于2020-04-07得票数 2
回答已采纳
所以当我先运行cuda.select_device(0)，然后运行cuda.close()。Pytorch不能再次访问图形处理器，我知道有办法让PyTorch可以再次使用图形处理器，而不必重新启动内核。但我忘了怎么做了。还有人知道吗？
from numba import cuda as cu
import torch
# random tensor
a=torch.rand(100,100)
#tensor can be loaded onto the gpu()
a.cuda()
device = cu.get_current_device()
device.reset()
#
浏览 1提问于2020-01-30得票数 10
我的colab会话总是在试图导入torch_geometric.data模块时崩溃。供参考，我正在编写的代码如下：
import torch
def format_pytorch_version(version):
return version.split('+')[0]
TORCH_version = torch.__version__
TORCH = format_pytorch_version(TORCH_version)
def format_cuda_version(version):
return 'cu' + version.r
浏览 17提问于2022-04-29得票数 0
回答已采纳
我正在尝试安装。不幸的是，最初的实现只支持Python1.7andPython0.4.0，所以我使用了一个包含了与torch 1.7 ( 2.7+ )兼容的修复程序的叉子。主要问题是使用AT_CHECK()，它与较新版本的PyTorch不兼容，而是被TORCH_CHECK()所取代。
在固定版本上运行pip install neural_renderer_pytorch之后，使用虚拟环境获得输出(我将输出截断为错误)：
/tmp/pip-install-[somestring]/neural-renderer-pytorch_[somelongstring]/neural_renderer/
浏览 24提问于2021-05-08得票数 0
回答已采纳
PyTorch和Torchvision用不同的CUDA版本编译。PyTorch拥有CUDA version=11.6和torchvision CUDA Version 11.3。请重新安装与您的PyTorch安装匹配的torchvision。
我多次尝试从网站重新安装torchvision，以及PyTorch和python。我被困住了，我不知道如何解决这个问题。
浏览 12提问于2022-11-16得票数 -1
2回答
在将我的环境的python升级到Python 3.61之后，我尝试使用以下命令安装pytorch：
conda安装-c peterjc123 pytorch
然而，我得到了这个错误：
正在获取包元数据.............解决软件包规范：。
UnsatisfiableError:发现以下规范冲突：
-pytorch -pyqt
我还使用了命令conda install -c peterjc123 pytorch cuda90 conda install -c peterjc123 pytorch cuda80
但结果仍然是一样的。有谁知道怎么解决这个问题吗？
浏览 3提问于2017-12-24得票数 1
在colab中，每当我们需要图形处理器时，我们只需单击change runtime type并将硬件加速器更改为GPU cuda可用，torch.cuda.is_available()就是True 如何做到这一点是AWS sagemaker，即打开cuda。我是AWS的新手，正在尝试在aws sagemaker中使用pytorch训练模型，Pytorch代码首先在colab环境中进行测试。 我的sagemaker笔记本电脑是ml.t2.medium
浏览 31提问于2021-04-02得票数 0
回答已采纳
我知道这个问题可能很愚蠢，但我在网上找不到任何帮助。最近我在我的电脑上安装了anaconda，它运行Windows10 x64。然后我使用anaconda提示符下载并安装了3.6python的pytorch：
conda install pytorch torchvision cuda100 -c pytorch
安装完成后，我在anaconda的提示符中验证是否安装了pytorch：
>>> Python
>>> Import torch
>>> torch.cuda.is_available()
True
我还检查了conda列表，确实
浏览 18提问于2018-12-13得票数 2
回答已采纳
我正在尝试在google colab中使用GPU。下面是我的colab中安装的pytorch和cuda版本的详细信息。 Torch 1.3.1 CUDA 10.1.243
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Sat_Aug_25_21:08:01_CDT_2018
Cuda compilation tools, release 10.0, V10.0.130 我对在pytorch模型上使用GPU进行迁移学习非常陌生。我的torch.cuda.is_
浏览 561提问于2019-12-03得票数 4
回答已采纳
我得用Docker捆绑我的系统。但是我的系统使用Java (JAR文件来运行)和PyTorch的python。我最初尝试使用openjdk:buster base映像，然后在其之上安装python3。所以JAR和PyTorch都能工作，但是PyTorch只支持CPU。但是现在我必须使用GPU加速我的PyTorch代码，为此我需要NVIDIA。在另一个码头，我发现nvidia/cuda:10.2-base-ubuntu18.04为我的PyTorch工作。但是这个码头工人不能运行JAR文件。
所以我被困在这两个组合中。我要么想
将NVIDIA-Cuda依赖项安装到openjdk码头基础映像安装op
浏览 2提问于2021-10-22得票数 1
我正在尝试运行一个需要pytorch-gpu的python包。我已经将我的Colab笔记本的运行时类型更改为GPU。当我运行该命令时，我会遇到以下错误。不确定我是否可以自己在colab上构建pytorch？
Traceback (most recent call last):
File "inference_unet.py", line 9, in <module>
import torchvision.transforms as transforms
File "/usr/local/lib/python3.6/dist-packages/
浏览 110提问于2020-05-07得票数 0
4回答
我已经安装了cuda和pytorch conda install pytorch torchvision cudatoolkit=10.0 -c pytorch 然而，看起来nvcc并没有随它一起安装。例如，如果我想使用nvcc -V，我得到的错误是找不到nvcc，并且我应该用sudo apt install nvidia-cuda-toolkit.安装它。我可以这样做吗(我不想只是尝试然后发现它不工作/弄乱整个cuda设置)。这是一个bug还是预期的行为？ 我使用的是Ubuntu 18.04和cuda 10.2
浏览 537提问于2019-06-06得票数 19
我正在做一个PyTorch几何项目，使用Google来支持CUDA。由于默认情况下它的库不存在，所以我运行：
!pip install --upgrade torch-scatter
!pip install --upgrade torch-sparse
!pip install --upgrade torch-cluster
!pip install --upgrade torch-spline-conv
!pip install torch-geometric
最近，由于版本升级而导入torch_geometric，时，出现了一个CUDA版本不匹配的情况：
RuntimeError:检
浏览 44提问于2020-02-15得票数 10
回答已采纳
早些时候，我已经配置了以下项目
在CUDA=10.2中使用Pytorch，现在CUDA=10.2支持的Pytorch在Windows中不可用。因此，当我使用Pytorch和CUDA=11.3配置同一个项目时，我会得到以下错误：
RuntimeError: Attempted to set the storage of a tensor on device "cuda:0" to a storage on different device "cpu". This is no longer allowed; the devices must match.
请帮我
浏览 89提问于2022-03-28得票数 0
回答已采纳
我的Ubuntu 16.04机器上安装了两个版本的CUDA : 9.0和10.1。它们分别位于/usr/local/cuda-9.0和/usr/local/10.1。如果我通过pip (pip install torch==1.6.0)安装PyTorch 1.6.0 (需要CUDA10.1)，它使用版本9.0，因此没有检测到GPU。我已经把我的LD_LIBRARY_PATH改成了"/usr/local/cuda-10.1/lib64:/usr/local/cuda-10.1/cuda/extras/CUPTI/lib64"，但PyTorch仍然在使用CUDA9.0。如何告诉P
浏览 40提问于2021-02-09得票数 2
回答已采纳
我的遥控器有cuda==11.0，我想在上面安装pytorch。
我使用命令conda install pytorch cudatoolkit=11.0 -c pytorch -c conda-forge，但在安装列表中：
cudatoolkit conda-forge/linux-64::cudatoolkit-11.0.3-h15472ef_8
pytorch pytorch/linux-64::pytorch-1.10.0-py3.8_cpu_0
我发现pytorch是一个cpu 。
或者，我将11.0替换为11.1，安装列表看起来如下：
cudato
浏览 9提问于2021-12-13得票数 3
回答已采纳
我正在尝试遵循facebook ()上聊天机器人的教程，但我被困在这里了: conda install pytorch torchvision cuda80 -c soumith我设法得到了除了cuda80之外的所有包，但没有找到。我也尝试过conda安装magma-cuda80 -c soumith conda安装cuda80 -c soumith conda安装magma-cuda80 -c pytorch。
但没有成功。PackagesNotFoundError:当前频道不提供以下软件包：- magma-cuda80
我在哪里可以找到包裹？我在Windows10上。
浏览 2提问于2018-02-20得票数 0
回答已采纳
我在Anaconda提示符中运行这段代码，它返回True。
(base) C:\User
torch.cuda.is_available()
True
但是当我运行其他conda环境时，它就是不起作用。
(pytorch_project) C:\User
torch.cuda.is_available()
False
问题似乎是torch.version.cuda.的不同结果
(基) torch.version.cuda = 10.1
( torch.version.cuda )pytorch_project= 10.2
但我不知道他们为什么不同..。
如何使10.2下降到10.1，并使is_
浏览 2提问于2020-10-22得票数 1
回答已采纳
每当我试图将一个变量移动到pytorch中的cuda (例如，torch.zeros(1).cuda() )时，我就会得到错误消息"AssertionError: Torch未通过启用CUDA编译“。此外，torch.cuda.is_available()还返回False。
我已经读过几个接近这个错误的答案，但是由于某些原因，几次试图重新安装cuda和putorch没有改变任何事情。下面是我使用的一些设置：
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
conda install pytorch torch
浏览 6提问于2022-01-14得票数 3
我正在使用cudas运行一些代码，我需要在CPU上测试相同的代码以比较运行时间。为了决定是使用常规pytorch张量还是使用cuda浮点张量，我使用的库称为torch.cuda.is _ 可用()。有没有简单的方法让这个函数返回false？我尝试使用以下命令更改Cuda可视设备 os.environ["CUDA_VISIBLE_DEVICES"]="" 但是torch.cuda.is _ available()仍然返回True。我查看了pytorch源代码，在我的例子中是torch.cuda.is _ 可获得的回报 torch._C._cuda_getDevi
浏览 350提问于2018-10-24得票数 4
回答已采纳
RuntimeError: Detected that PyTorch and torchvision were compiled with different CUDA versions. PyTorch has CUDA Version=10.2 and torchvision has CUDA Version=10.1. Please reinstall the torchvision that matches your PyTorch install. 我试图在我的Google Colab上运行YOLACT，发现了这个错误。有人能帮助解决这个问题吗？
浏览 596提问于2020-07-24得票数 2
我怎样才能找到在我的电脑上安装在Windows 10上的PyTorch 1.13的数据自动化系统11.x？
我尝试过:我从NVIDIA网站上安装了NVIDIA CUDA驱动程序和Windows工具包。我可以通过在木星实验室中输入：!nvidia-smi来验证这一点，这给了我以下输出。这表明已经安装了CUDA工具，但我的PyTorch包没有使用。我需要了解什么版本的CUDA驱动程序被安装，以便我可以安装正确的PyTorch-GPU软件包。
+-----------------------------------------------------------------------------+
浏览 22提问于2022-11-13得票数 -1
我已经建立了一个带有Pytorch和GPU支持的CNN。它在python中运行良好，我想使用VS2017将它导入到我的C++应用程序中。我正在遵循和的说明。
我可以构建和运行应用程序，但在加载模型时出现以下错误：
error loading the model
Cannot initialize CUDA without ATen_cuda library. PyTorch splits its backend into two shared
libraries: a CPU library and a CUDA library; this error has occurred becaus
浏览 14提问于2019-10-07得票数 0
导入torchvision时出现以下错误。 Detected that PyTorch and torchvision were compiled with different CUDA versions. PyTorch has CUDA Version=11.0 and torchvision has CUDA Version=10.1. Please reinstall the torchvision that matches your PyTorch install. 如何将pytorch的cuda版本改为10.1？ ‘'conda install pytorch torch
浏览 322提问于2020-12-28得票数 2
回答已采纳
我已经用pip和conda安装了pytorch cuda 当我在IDLE中运行此命令时： >> import torch
>> torch.cuda.is_available() 我得到了"True"，但在Spyder或Jupyter Notebook中，即使在更新了包和conda之后，它也给出了"False“。pytorch cuda的conda更新从10.1升级到10.2
浏览 46提问于2020-06-17得票数 2
1回答
具有CUDA功能的 NVIDIA GeForce RTX 3070与当前的PyTorch安装不兼容。当前的PyTorch安装支持CUDA功能sm_37 sm_50 sm_60 sm_70。
因此，我目前正试图训练一个神经网络，但我明白这个问题。我所拥有的GPU模型似乎与我所拥有的PyTorch版本不兼容。
我的nvcc -V的输出是：
Cuda compilation tools, release 11.4, V11.4.48
Build cuda_11.4.r11.4/compiler.30033411_0
我的PyTorch版本是1.9.0。
我尝试过将CUDA版本从最初的10版本更改为
浏览 0提问于2021-07-26得票数 3
我有一台上网速度慢的电脑。当试图安装一个大约500 to的包(conda install -c pytorch pytorch)时，我会得到一个下载超时。
我尝试将该文件手动下载到其他地方，然后将其放在我的主目录中并使用conda install --offline ~/pytorch-1.0.0-py3.7_cuda9.0.176_cudnn7.4.1_1.tar.bz2进行安装(如这里)。然而，当我试图安装更多依赖于它的软件包时，conda决定下载它：
The following packages will be UPDATED:
pytorch: 1.0.0-py
浏览 0提问于2019-01-04得票数 2
回答已采纳
1回答
我尝试使用cli进行推理，从检测和识别模型中获得预测。用cuda10.2完成推理需要15分钟，但是我有cuda11.3，它需要3个小时，我想减少这一次。注意:我的硬件不支持cuda10.2。
因此，我安装了以下软件包，
h2bc3f7f_2pytorch 11.3.1 cudatoolkit 11.3.1 py3.7_cuda11.3_cudnn8.2.0_0 1.10.0 py3.7_cuda11.3_cudnn8.2.0_0 0.11.0 py37_cu113 cudatoolkit
当我运行推论cli时，我得到了这个错误，
RuntimeError: CUDA out of memory
浏览 3提问于2022-07-12得票数 1
1回答
我想安装PyTorch，为此我访问了PyTorch官方网站，他们命令我用Cuda安装它：
pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
他们希望我为PyTorch安装的Cuda版本是11.1。但是我已经在我的计算机上安装了Cuda 11.2 (用于TensorFlow 2.5.0)。我的问题是，如果我用他们给我的命令安装PyTorch，它会删除Cuda11.2吗？如果它删除了Cud
浏览 0提问于2021-07-14得票数 0
回答已采纳
2回答
我使用下面的命令在我的conda环境中安装py手电筒。
conda install pytorch=0.4.1 cuda90 -c pytorch
但是，我得到了以下错误
解决环境:失败
PackagesNotFoundError:以下软件包无法从当前频道获得：
pytorch=0.4.1
cuda90
现有渠道：
若要搜索可能提供所查找的conda包的备用通道，请导航到
我怎么才能解决这个问题？我当然安装了cuda 9和nvcc工厂。
浏览 4提问于2020-08-05得票数 2
回答已采纳
我正在尝试安装PyTorch与数据自动化系统。我遵循了中提到的说明(使用conda安装)
conda安装手电筒cudatoolkit=11.3 -c手电筒
conda install命令运行时没有出现任何错误：
conda list显示以下内容：
# Name Version Build Channel
cudatoolkit 11.3.1 h2bc3f7f_2
pytorch 1.11.0 py3.9_
浏览 24提问于2022-05-06得票数 1
回答已采纳
1回答
我用cuda 7.5通过conda安装了火把
conda install pytorch=0.3.0 cuda75 -c pytorch
>>> import torch
>>> torch.cuda.is_available()
True
除了这个之外，我没有为cuda做任何其他的安装，因为它看起来好像是随cuda一起来的。
现在，我正在尝试设置yolo2
但是，我在./make.sh命令中出现了错误
这是错误
OSError:无法在您的$PATH中找到nvcc二进制文件。要么将其添加到您的路径中，要么设置$CUDAHOME
我假设我需要在我的
浏览 0提问于2018-09-12得票数 4
我刚接触pytorch geometric，尝试将其安装到我的计算机上，但失败了，所以我尝试在Google Colab上运行代码。根据前面的问题(这对我没有帮助，我也不确定是不是同一个问题)： PyTorch Geometric CUDA installation issues on Google Colab 我做到了： !pip install --upgrade torch-scatter
!pip install --upgrade torch-sparse
!pip install --upgrade torch-cluster
!pip install --upgrade t
浏览 167提问于2020-04-19得票数 4
2回答
不能将GPU与火炬一起使用
、、、、
我一直在尝试使用Py手电筒时出现这个错误。
RuntimeError:试图在CUDA设备上反序列化对象，但是torch.cuda.is_available()是假的。如果您在一台只运行CPU的机器上运行，请使用torch.load和map_location=torch.device(“CPU”)将您的存储映射到CPU。
我使用conda install pytorch torchvision cudatoolkit=10.1 -c pytorch安装了Pytorch。
有了tensorflow，我的GPU运行得很好。
浏览 3提问于2019-12-27得票数 3
直到2022年的Apr26th，CUDA已经更新为11.6版本，该版本可以由安装
wget https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux.run
sudo sh cuda_11.6.2_510.47.03_linux.run
我想cudatoolkit的版本也将是11.6
然而，没有一个版本的pytorch与CUDA11.6相匹配。
在的网站上，最新的CUDA版本是11.3，pytorch版本将是
conda install pytor
浏览 23提问于2022-04-26得票数 2
回答已采纳
torch.cuda.is_available()返回False。使用以下方法安装PyTorch：
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
已经尝试安装CUDA 11.7和11.8了。环境详情：
2080sNVIDIA-SMI RTX 522.06驱动程序版本: 522.06CUDA版本: 11.8
来自pip list的版本
torch 1.13.0
torchaudio 0.13.0
torchinfo 1.7.1
浏览 5提问于2022-11-17得票数 -5
我在机器中添加了一个GeForce GTX1080Ti(运行Ubuntu18.04和使用Python3.7运行Anaconda )，以便在使用PyTorch时使用GPU。这两张卡片的识别都是正确的：
$ lspci | grep VGA
03:00.0 VGA compatible controller: NVIDIA Corporation GF119 [NVS 310] (reva1)
04:00.0 VGA compatible controller: NVIDIA Corporation GP102 [GeForce GTX 1080 Ti] (rev a1)
NVS 310处理我的2
浏览 5提问于2019-01-19得票数 13
我之前已经从C++推断出了TensorFlow图。现在我开始研究如何通过C++推断PyTorch图。 我的第一个问题是，我如何知道推荐的cuDNN版本与LibTorch一起使用，或者我是否正在进行自己的PyTorch编译？ 确定推荐的CUDA版本很容易。转到https://pytorch.org/并选择Quick Start Locally下的选项(PyTorch Build、Your OS等)该网站明确表示推荐CUDA10.1，但没有提到cuDNN版本，在谷歌搜索后，我找不到这个问题的确切答案。 根据我对ubuntu上的PyTorch的了解，如果你使用Python版本，你必须安装CUDA驱
浏览 45提问于2020-02-02得票数 1
回答已采纳
我需要使用具有cuda gpu访问权限的Pytorch。我使用"pytorch/pytorch:1.9.0-cuda10.2-cudnn7-runtime“中的作为我的基础图像。必要的python版本是否也随此命令一起安装？
浏览 47提问于2021-07-30得票数 0
我使用conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge命令安装了带有cuda支持的pytorch。但是，当我运行我的项目代码时，`torch.cuda.is_available()‘返回False并导致错误。但是，当我在项目外部(在相同的环境中)检查时，它会返回True。我把截图加在下面。
规范: Windows 10，Python3.9.6，Py火炬1.9.0，CUDA 11.1
你知道是什么导致了这个错误吗？我该怎么解决呢？提前谢谢。
浏览 0提问于2021-07-14得票数 0
我在我的Ubuntu 16.04系统上有一个conda环境。 使用以下命令安装Pytorch时： conda install pytorch 然后我尝试运行我需要的脚本，我得到了错误消息： raise AssertionError("Torch not compiled with CUDA enabled") 通过查看论坛，我发现这是因为我安装的Pytorch没有CUDA支持。 然后我试着： conda install -c pytorch torchvision cudatoolkit=10.1 pytorch 但现在我得到了错误： from torch.utils
浏览 53提问于2020-01-02得票数 2
回答已采纳
扫码
添加站长 进交流群
领取专属 10元无门槛券
手把手带您无忧上云
相关资讯
热门标签
云服务器
ICP备案
对象存储
即时通信 IM
实时音视频活动推荐
腾讯云开发者
