PyTorch的Cuda和Tensorflow的Cuda

Question

我想安装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吗？如果它删除了Cuda11.2并将其替换为Cuda11.1，那么TensorFlow将无法在GPU上运行。为了使用TensorFlow，我必须牺牲PyTorch吗？有什么办法让他们两者兼得吗？

Answer 1

解决问题的一种方法是创建虚拟环境。下面是两种建立虚拟环境的方法

• pip虚拟环境
• conda虚拟环境

我建议为不同版本的Tensorflow、Py火炬、CUDA建立anaconda (conda)虚拟环境。

遵循这个指南，tensorflow和Py手电筒conda指南