参考链接:
给虚拟环境指定cuda
本文重述原作者内容,主要是防止原链接丢失。
服务器上装了多个版本的cuda,有的时候需要使用TensorFlow,有的时候需要使用pytorch,有时候需要使用cuda9.0+,有的时候需要使用cuda10.0+。那么必然涉及到给每个不同的虚拟环境配上不同的cuda版本。
本文以有cuda9.0 和 cuda10.1为例,介绍如何给pytorch这个环境指定cuda10.1版本,启动环境时自动执行脚本激活cuda10.1。
1.首先激活虚拟环境 pytorch,输出当前环境的路径
conda activate pytorch
echo ${
CONDA_PREFIX}
得到路径, for example:/home/username/anaconda3/envs/pytorch
2.建立激活环境下的文件夹,写入脚本
#建立激活环境下的文件夹
mkdir -p /home/username/anaconda3/

最低0.47元/天 解锁文章
322

被折叠的 条评论
为什么被折叠?



