cuda不同版本切换1.3

最近也是学了一点,现在更新一下。之前一直是在自己电脑下载好几个版本的cuda,同时还是添加环境变量特别麻烦,最近才知道在虚拟环境中也可以下载,特此更新一下。我的baseline比较老,我的版本信息是TensorFlow-gpu1.4-Python3.5-cuda8.0-cudnn6.0,不知道的伙伴可以去TensorFlow官网查询自己所需要的对应版本信息。

1.使用anconda来管理虚拟环境,建立虚拟环境,djw我是随便取的

conda create -n djw python=3.5

2.进入虚拟环境

conda activate djw

3.安装cuda和cudnn,伙伴们可以根据自己的需求去设置

conda install cudatoolkit=8.0
conda install cudnn=6.0

4.因为TensorFlow版本很老,我选择离线安装,然后conda install会出现识别路径的问题,不是很稳定,因此建议用pip安装,如果不确定要安装的版本是否在conda库里面被找到,可以使用搜索指令

conda search tensorflow-gpu

 5.离线安装

cd D:\ #进入安装包目录
pip install D:\tensorflow_gpu-1.4.0-cp35-cp35m-win_amd64.whl 

6.等待安装完成,退出虚拟环境

conda deactivate

7.检查虚拟环境是否建立成功,并根据显示位置信息,在使用pycharm选取编译器时候将虚拟环境添加即可

conda env list

8.我也是成功开始训练了

 

  • 4
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值