win10和Nvidia显卡(1050Ti及最新驱动)
#命令创建python版本为X.X、名字为your_env_name的虚拟环境。
conda create -n your_env_name python=X.X(2.7,3.6,3.7等)
conda remove -n your_env_name --all
activate your_env_name
conda install tensorflow-gpu
pip install tensorflow-gpu -i https://pypi.douban.com/simple
pip install -U tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple
安装过程默认yes
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
tf.test.is_gpu_available()
## tensorflow 2.0
import tensorflow as tf
sess=tf.compat.v1.Session(config=tf.compat.v1.ConfigProto(log_device_placement=True))
conda activate tensorflow
conda install ipykernel
python -m ipykernel install --user --name tensorflow --display-name "Python (python36)"
其中 tensorflow 为自己想在 jupyter 运行的环境,后面的 Python (python36) 为新的 kenel 的名字,然后切换到自己想工作的目录下执行 jupyter 命令,目录的切换不多说
jupyter notebook
或者
jupyter lab
点击Kernel,选择Change Kernel
选择自己的 kernel 环境,kernel的位置有的会不一样,选择刚刚自己设置好的kernel