anoconda环境中pytorch和tensorflow环境的兼容
问题:pytoch和tensorflow的cuda版本如果不一致,但是.bashrc中只能一次设置一个cuda路径。每次激活tensorflow或者pytorch的虚拟环境时,要重新去.bashrc文件中添加对应环境的cuda路径,这样非常的麻烦。 解决方法: 一、每次使用不同cuda版本的环境时,手动修改.bashrc文件中对应环境的cuda路径。 1.激活tensorflow环境时(cuda6) 打开.bashrc文件,在最后写上cuda的路径,保存并退出: export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/mnt/cuda_libs/cudnn/v6.0
用户评论