conda配置鏡像並安裝gpu版本pytorch和tensorflow2

一、安裝conda

 

 

 

 

 

 二、安裝CUDA

1、查看顯卡型號：我的電腦——》管理---->設備管理器——》顯示適配器，可以看到GTX1060

 

 2、下載相應的控制面板

 

 3、查看控制面板：控制面板-》硬件和聲音-》NVIDIA控制面板，左下角系統信息，組件。

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

注意，路徑不要變

 

一、刪除鏡像

# 換回默認源（清除所有用戶添加的鏡像源路徑，只保留默認的路徑）
conda config --remove-key channels
# 顯示所有鏡像通道路徑命令 
conda config --show channels
#刪除環境    conda remove -n rcnn --all

二、添加鏡像

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch

conda config --set show_channel_urls yes

三、創建環境

注意，只能下載python3.7以下。如pytorch1.2，torchvision0.4,python3.7cuda10.0

conda clean -i
conda create -n pt_gpu python=3.7
activate pt_gpu

四、安裝

conda install pytorch torchvision cudatoolkit=10.0
pip install sentence_transformers
pip install faiss-cpu
pip install tensorflow-gpu==1.14.0
或者
pip install tensorflow==1.13.0rc1 transformers sentence_transformers faiss-cpu

 五、驗證

import torch
print(torch.__version__)
print("gpu:", torch.cuda.is_available())

 六、安裝tensorflow

conda create -n tf2 python=3.7
conda activate tf2

pip install tensorflow-gpu==2.0.0-rc0

import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'

import tensorflow as tf

a = tf.constant(1.)
b = tf.constant(2.)
print(a+b)

print('GPU:', tf.test.is_gpu_available())