pytorch环境配置

2019-07-17  本文已影响0人  renew_c376

anaconda安装:

1.进入anaconda官网Home - Anaconda 下载anaconda

2.安装anaconda,注意,可以将anaconda加入环境变量中,这样在cmd命令行中直接输入conda命令:

图1

pytorch安装:

1.进入pytorch官网:PyTorch

图二

复制红线勾出的,粘贴到cmd中进行安装就行了。

安装好后,在anaconda 的spyder中输入:import torch可以进行检验是否安装成功。

cuda安装:

1.进入cuda官网:CUDA Toolkit 10.1 Update 1 Download | NVIDIA Developer

图3

这里以window10为例子,installer type为:exe(local)(注:显卡为N卡的才能安装cuda)

安装好了可以进行检验:

在cmd中输入nvcc -V,如果出现以下命令则说明安装成功(见图4):

图4

验证pytorch是否可以用gup:

在anaconda 的spyder下输入:torch.cuda.is_available() 则可以验证cuda是否可用,如下图:

图5

百度网盘:

https://pan.baidu.com/s/1jN1-rJ1-Lu4xmT6Ijtd8rQ 密码:rkab

百度网盘中包括win10的pytorch whl文件,cuda 10.1文件

上一篇 下一篇

猜你喜欢

热点阅读