简介:在运行不同的代码时,可能会用到不同版本的Pytorch/Tensorflow,不同版本的Pytorch/Tensorflow在使用GPU时,对应不同的CUDA版本。在系统中安装多个版本的CUDA比较麻烦,这时候就可以在虚拟环境里面直接安装CUDA。
在Anaconda虚拟环境中安装CUDA,可以按照以下步骤进行操作:
创建新环境:使用conda命令创建一个新的虚拟环境。命令如下:
conda create —name
其中,
conda create —name myenv tensorflow=2.3.0=py38_cuda10.1
上述命令将创建一个名为“myenv”的虚拟环境,并安装Tensorflow 2.3.0和对应版本为10.1的CUDA。
激活虚拟环境:使用conda命令激活已创建的虚拟环境。命令如下:
conda activate
例如:
conda activate myenv
上述命令将激活名为“myenv”的虚拟环境。
验证安装:在虚拟环境中运行代码,检查CUDA是否成功安装。对于PyTorch,可以运行以下代码:
python -c “import torch; print(torch.cuda.is_available())”
如果输出为True,则表示PyTorch成功识别到CUDA。对于Tensorflow,可以运行以下代码:
python -c “import tensorflow as tf; print(tf.test.is_gpu_available())”
如果输出为True,则表示Tensorflow成功识别到CUDA。