pytorch安装(python3.7 cuda9.2)

先打开Anaconda Prompt

新建一个环境,可以取名为pytorch:

conda create --name pytorch python=3.7 anaconda

python版本根据你自己的python版本所设

我首先考虑用的anaconda安装(建议先改源,如清华镜像源),按讲应该去PyTorch下找到对应的安装指令

我cuda9.2,要我去github上找指令?WTF?

网上找的conda安装的指令貌似不行。

只好用pip安装了

pip install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp36-cp36m-win_amd64.whl

注意这里cu92如果你cuda是9.0就是cu90,如果是8.0就是cu80.

但是安装要先下载,下载太慢了啊,经常因为各种原因重新下载。这时你可以浏览器中直接网址访问那个地址,把文件下到本地,然后本地安装,速度很快,我就是这么做的。

例如我是下到本地后移到D:/目录下直接安装。安装成功后尝试输入

import torch

print(torch.__version__)

如果不报错,就是安装成功。

最后需要安装 torchvision

我一开始用的是网上找的指令,

pip3 install torchvision

后来发现不对。

不能找到一个满足需要的版本?

后来发现上面不是说要安装0.4.1的版本吗,这个版本是0.5.0的了。

于是改为如下指令:

pip install torchvision==0.4.1

安装成功!

到这一步还没有完,对于使用iPython、jupyter notebook、jupyter lab的小伙伴,如果就这样打开如jupyter lab,执行import torch还是会报错。

你需要装nb_conda或者ipykernel。我两个都安装了,但ipykernel这条路我只尝试了一半,详见参考文档。

安装nb_conda:

1、进入虚拟环境pytorch

conda activate pytorch

2、在pytorch虚拟环境中安装插件nb_conda

conda install nb_conda

安装 ipykernel:

1、进入虚拟环境

conda activate pytorch

2、安装 IPykernel

conda install ipykernel

3、将虚拟环境加入IPykernel

python -m ipykernel install --user --name=pytorch

安装好之后,要先激活pytorch环境,再打开jupyter lab,否则还是没法import torch。

例如我的jupyter lab要在“D:/人工智能”这个目录下打开,那么每次打开都要在cmd下依次执行以下指令:

cd /d d:\人工智能
activate pytorch
jupyter lab

这样岂不麻烦。

于是可以写一个bat文件,打开记事本,输入以下指令:

call activate pytorch
call jupyter lab

把它保存在“D:/人工智能”这个目录下,重命名后缀名为“.bat”,生成快捷方式放在桌面。这样每次直接在桌面启动就行了。

启动顺利,并且可以选择环境。

大功告成!

参考博客:

pytorch 0.4.1+win10 +python3.66+cuda9.2 安装

Could not find a version that satisfies the requirement torch==1.4.0 (from torchvision)

Jupyter Notebook 中添加 conda 虚拟环境

jupyter notebook中添加conda虚拟环境

  • 2
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值