本文主要是介绍小米游戏本基于anaconda安装pytorch(RTX2060+win10),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
用的是小米游戏本 RTX2060
基于anaconda,主要参考这一篇:
https://www.cnblogs.com/zhouzhiyao/p/11784055.html
1、进入命令行,查看cuda版本
2、安装cudnn,教程很多,要注册账户,填调查问卷。解压缩,复制粘贴即可,一般不会出大问题。
参考:https://blog.csdn.net/zhangpengzp/article/details/85008507
3、装python3.7
4、装pytorch,用国内的镜像源,不然速度太慢
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/
conda install pytorch torchvision cudatoolkit=10.1
5、测试
6、安装pychram,用社区版就行
7、在pycharm中选用conda环境
8、在pycharm环境中运行torch程序,我用的是莫烦python-pytorch例程。import torch语句没有报错,大功告成。
9、安装一些库的方法
(1)可以在anaconda navigator中选择直接下载;
(2)可以在anaconda prompt中安装,比如sklearn.
这篇关于小米游戏本基于anaconda安装pytorch(RTX2060+win10)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!