GeForce GTX 920M安装Pytorch

创建python3.6的环境

conda create -n python36 pip python=3.6 

激活环境

conda activate python36 

添加镜像源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/    
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ 
conda config --set show_channel_urls yes                                                  
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

退cuda版本,同时降低pytorch版本,这里给出方案是,使用cuda10.0和pytorch1.2.0

conda install pytorch torchvision cudatoolkit=10.0 
pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html

最后别忘记,测试一下

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

推荐阅读更多精彩内容