使用 --cuda 选项来指定 Conda 中应该使用的 CUDA 版本。
时间: 2024-03-07 17:49:29 浏览: 76
switch-cuda:一个简单的bash脚本,用于在已安装的CUDA版本之间进行切换
是的,你说得对。如果你在创建 Conda 虚拟环境时想要指定使用的 CUDA 版本,可以使用 `--cuda` 选项。
例如,如果你想要在创建名为 `my_env` 的 Conda 虚拟环境时使用 CUDA 10.1,可以使用以下命令:
```
conda create --name my_env --cuda 10.1
```
这会在 `my_env` 中安装 CUDA 10.1,而不是继承系统中已安装的 CUDA 版本。需要注意的是,你必须在系统中安装了所需版本的 CUDA 才能使用该版本。
希望这能帮助你解决问题!
阅读全文