CUDA9.1:Pytorch GPU版安装

以下命令在Anaconda Prompt中执行。

一、创建虚拟环境、切换虚拟环境

cuda9.1对于的python版本是3.6

#conda create -n 虚拟环境名 python=版本号
#创建一个python版本是3.6的python环境,命名为pytorch-gpu
conda create -n pytorch-gpu python=3.6

#删除虚拟环境
conda remove -n pytorch-gpu --all

#创建完以后进入虚拟环节
conda activate pytorch-gpu

#查看所有虚拟环境
conda-env list

#退出环境
conda deactivate

#列出所有环节,可以使用它查看环节是否安装成功
conda info --envs

#查看conda镜像配置
conda config --show

#清除索引缓存,保证镜像更新
conda clean -i

二、安装GPU版pytorch

conda安装对应镜像地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64

pip安装对应地址(.whl):

https://download.pytorch.org/whl/cu90/torch_stable.html

## pytorch
pip3 install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-win_amd64.whl 

## torchvision
pip install torchvision==0.1.8 

三、安装jupter notebook

activate pytorch-gpu

conda install ipykernel

#--name后写虚拟环境名称 --display-name后面写jupter展示名称
python -m ipykernel install --user --name pytorch-gpu --display-name pytorch-gpu


conda install nb_conda    #安装以后在开始菜单中看到对应的jupter notebook

四、验证是否安装成功

#安装numpy激活环境
conda install numpy

#测试是否安装成功
打开jupyter notebook用GPU环境(这里是pytorch-gpu)创建.ipynb文件,输入以下语句
import numpy
import torch
import torchvision
print(torch.cuda.is_available())#输出为true即可

如果不出结果且重命名显示:error,可以通过下面方法解决

 pip install pyzmq==19.0.2

#如果装完后会跳出红字error:pip's dependency...提示需要pyqt5和pyqtengine并不大于5.13版本继续输入
pip install PyQt5==5.12 -i https://pypi.douban.com/simple
#装完后再装
pip install pyqtwebengine==5.12 -i https://pypi.douban.com/simple

五、Jupyter Notebook无法跳转到浏览器的问题

jupyter notebook --generate-config

C://User//用户名下生成一个.jupyter,打开里面的jupyter_notebook_config 文件:用文本打开,找到 #c.NotebookApp.brower = ’ ,将其进行如下修改:

import webbrowser
webbrowser.register("chrome",None,webbrowser.GenericBrowser("C:\\Program Files(x86)\\Google\\Chrome\\Application\\chrome.exe"))
c.NotebookApp.browser = 'chrome'

这里我默认用谷歌浏览器,括号里是我的谷歌浏览器地址,你可以根据自己喜好改变。

六、jupyter notebook 更改默认存储路径和在指定文件夹下打开的方法

打开里面的jupyter_notebook_config 文件

找到这个位置,输入自己想打开的起始路径。

此时,如果你还没有成功,在win开始菜单中找到jupyter notebook快捷图标,鼠标右键>>更多>>打开文件位置>>目标,然后右击Jupyter Notebook(anaconda)文件,打开属性,删除目标里的"%USERPROFILE%/"这一行语句,加入自己的地址然后应用关闭即可。


 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值