1. 首页
  2. 数据库
  3. 其它
  4. anoconda环境中pytorch和tensorflow环境的兼容

anoconda环境中pytorch和tensorflow环境的兼容

上传者: 2021-02-01 16:15:58上传 PDF文件 32KB 热度 22次
问题: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
用户评论