Win10下用Anaconda3安裝TensorFlow和pytorch(python3.5)
阿新 • • 發佈:2019-01-08
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))