如何告诉PyTorch采用哪个CUDA版本？
要告诉PyTorch采用哪个CUDA版本，可以通过设置环境变量来实现。具体步骤如下：
- 首先，确保已经安装了所需的CUDA驱动程序和对应的CUDA工具包。
- 打开终端或命令提示符，输入以下命令来设置环境变量：
- 打开终端或命令提示符，输入以下命令来设置环境变量：
- 注意：上述命令中的
x.x应替换为实际的CUDA版本号，例如10.2。 - 接下来，可以通过以下方式告诉PyTorch采用哪个CUDA版本：
- 在Python代码中设置环境变量：
- 在Python代码中设置环境变量：
- 或者在终端或命令提示符中使用
CUDA_VISIBLE_DEVICES环境变量： - 或者在终端或命令提示符中使用
CUDA_VISIBLE_DEVICES环境变量： - 这样，PyTorch将会使用指定的CUDA版本进行计算。
关于PyTorch和CUDA的更多信息，可以参考腾讯云的相关产品和文档：
- PyTorch产品介绍：https://cloud.tencent.com/product/pytorch
- CUDA产品介绍：https://cloud.tencent.com/product/cuda
相关·内容
我的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/loca
浏览 40提问于2021-02-09得票数 2
回答已采纳
我正在研究如何在PyTorch中的不同节点上与多个GPU进行点对点通信。
从1.10.0版本开始，用于的PyTorch页面表示send的问号，而带有MPI后端的GPU的recv则表示问号。如果有人成功地设置了PyTorch，以便torch.distributed允许在多个GPU上进行点对点通信，请告诉我以及您是如何设置的。具体而言，您使用的是哪个MPI？那么pyTorch和Cuda的版本呢？
浏览 13提问于2021-12-17得票数 2
由于某种原因，我不得不使用cuda版本10.0而不是升级它，DriverAPI的版本比RunTimeAPI高，但有人告诉我没关系，其他问同样问题的人最后发现他们的版本不匹配。emmmm不喜欢我操作系统:Windows10 Python 3.7和3.8都尝试过nvidia-smi: NVIDIA-SMI451.67驱动程序版本: 451.67 CUDA<
浏览 2提问于2020-11-20得票数 0
/__init__.py:104: UserWarning: 这是令人困惑的，因为它指向通常的pytorch</e
浏览 12提问于2021-04-08得票数 9
PyTorch未启用和访问GPU驱动程序和CUDA。torch.cuda.is_available()返回falseimport torch它返回false。
因此，我按照Pytorch上的说明安装了Anaconda和Cuda
浏览 8提问于2019-09-16得票数 2
1回答
哪一个是命令来查看conda中的pytorch所看到的“正确”的CUDA版本？是一个类似的问题，但并没有让我走得太远。nvidia-smi说我有cuda版本的10.1torch.cuda.is_available()显示为假，所以它看到了No CUDA_cuda_getCompiledVersion(), 'cuda
浏览 3提问于2020-09-27得票数 26
回答已采纳
是否可以确定正在使用哪个已安装版本的cuda installed tensoflow？ 例如，在pytorch中我可以做到：torch._C._cuda_getDriverVersion() 注意:我不是在寻找像Which TensorFlow and CUDA version combinations are compatible?
浏览 16提问于2019-03-24得票数 0
回答已采纳
我如何通过终端或导航器使用Anaconda从1.4 -> 1.5更新PyTorch？在PyTorch Github页面上有以下命令
conda install -c pytorch magma-cuda90# or [magma-<e
浏览 0提问于2020-04-07得票数 2
回答已采纳
当我在colab GPU中运行一个python脚本时，我得到了以下错误：
Pytorch版本: 1.1.0，Cuda版本: 9.0。
如果需要任何其他细节，请告诉我。
浏览 3提问于2021-06-28得票数 1
我试图在conda下安装特定的PyTorch版本：pip3 install pytorch==1.0.1ERROR: Could not find a version that satisfies the requirement pytorch==1.0.1 (from versions: 0.1.2, 1.0.2)
ERROR: No matching
浏览 5提问于2020-09-25得票数 1
回答已采纳
我在CUDA (Linux上)上安装了pytorch的方法是：
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu20
浏览 13提问于2022-01-06得票数 3
回答已采纳
我需要在google colab中更改pytorch版本，所以我安装了anaconda %%bashtorch.__version__ 我是1.9+cuda120 更重要的是，当我试图 pip uni
浏览 579提问于2021-08-24得票数 2
我现在的PyTorch版本是1.10。我想把它降级到1.3，但是conda建议只使用CPU版本。我的GPU是一个GTX 1080 TI。我现在的设置是cudnn 7.6.5 cuda10.2_0
pytorch 1.10.0 py3.7_cuda10.2_cud
浏览 15提问于2021-12-07得票数 0
回答已采纳
下面的依赖链存在问题：
如何安装Visual 15.4？如何告诉
浏览 1提问于2019-06-25得票数 1
回答已采纳
5回答
在一个项目中，我们使用PyTorch。如何将其添加到poetry中似乎没有明显的方法来决定要安装哪个PyTorch版本。我考虑将不同的安装指令添加为额外的依赖项，但找不到一个选项来获得类似的设置：
pip3 install torch==1.3
浏览 62提问于2019-12-03得票数 30
回答已采纳
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的
浏览 322提问于2020-12-28得票数 2
回答已采纳
现在我开始研究如何通过C++推断PyTorch图。 我的第一个问题是，我如何知道推荐的cuDNN版本与LibTorch一起使用，或者我是否正在进行自己的PyTorch编译？确定推荐的CUDA版本很容易。转到https://pytorch.org/并选择Quick Start Locally下的选项(PyTorch Build、Your OS等)该网站明确表示推荐CUDA10.1，但没有提
浏览 45提问于2020-02-02得票数 1
回答已采纳
PyTorch和Torchvision用不同的CUDA版本编译。PyTorch拥有CUDA version=11.6和torchvision CUDA Version 11.3。请重新安装与您的PyTorch安装匹配的torchvision。
我多次尝试从网站重新安装torchvision，以及PyTorch和python。我被困住了，我不知道如何解决这个问题。
浏览 12提问于2022-11-16得票数 -1
我如何将一个GPU分配给Tensorflow模型，另一个GPU分配给Py火炬模型？设置CUDA_VISIBLE_DEVICES=0,1只告诉这两个模型这些GPU是可用的--我(在Python猜中)，如何确保Tensorflow采用GPU 0，Pytorch采用GPU 1？
浏览 0提问于2018-09-11得票数 0
回答已采纳
其中一个问题是：换句话说:我可以使用NVIDIA "cuda工具包“来安装火把吗？上下文：conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
接受"None“
浏览 2提问于2020-07-29得票数 3
回答已采纳
扫码
添加站长 进交流群
领取专属 10元无门槛券
手把手带您无忧上云
相关资讯
热门标签
云服务器
ICP备案
实时音视频
对象存储
即时通信 IM活动推荐
腾讯云开发者
