pytorch环境配置

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文件

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容