1. 程式人生 > >Win10下用Anaconda3安裝TensorFlow和pytorch(python3.5)

Win10下用Anaconda3安裝TensorFlow和pytorch(python3.5)

1、安裝Anaconda3。

Anaconda預設Python3.6,由於TensorFlow需要Python3.5,因此需要在root環境下更改Python版本。

2、新增映象源。

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 --set show_channel_urls yes

3、安裝Python3.5。

conda install python=3.5

等一會。

4、升級pip。直接用pip安裝TensorFlow會出錯。

使用管理員許可權開啟Anaconda Prompt

pip install --upgrade pip

關閉Anaconda Prompt,重新開啟。

5、安裝tensorflow。(CPU版本)

pip install --upgrade --ignore -installed tensorflow  

6、測試tensorflow。

spyder

在Spyder中輸入:

import tensorflow as tf  
hello = tf.constant('Hello, TensorFlow!'
) sess = tf.Session() print(sess.run(hello))

7、安裝pytorch。

先安裝

conda install numpy mkl cffi

或者需要升級mkl庫

pip install mkl
pip install mkl --upgrade

或者

conda install mkl
conda update mkl

如果想要更新所有庫

conda update --all

安裝pytorch。(CPU版本)
連結: https://pan.baidu.com/s/10K8x6n2a51hKKD5sc3_9YA 密碼: mn7r
下載 pytorch-cpu-0.3.1-py35_cpuhe774522_2.tar.bz2

conda install --offline pytorch-cpu-0.3.1-py35_cpuhe774522_2.tar.bz2
# 測試 tensorflow 和 pytorch是否正常工作
import tensorflow as tf  
import torch
from torch.backends import cudnn

hello = tf.constant('Hello, TensorFlow!')  
sess = tf.Session()  
print(sess.run(hello))

x = torch.Tensor([1, 3])
y = torch.Tensor([2, 4])
z = x + y
# z = z.cuda()    # cpu版本沒有cuda  
print(z)
print(cudnn.is_acceptable(x))