Windows 10系统在Anaconda下安装GPU版Pytorch

前置环境:

windows10+anaconda3+python3.7+cuda9.1.104+cudnn7.4.2

受限于Cuda版本,暂时安装pytorch v1.1.0

语句命令为:conda install pytorch==1.1.0 torchvision==0.3.0 -c pytorch

测试:

 

 

安装完毕

 

posted @ 2021-06-14 11:51  折风  阅读(87)  评论(0编辑  收藏  举报