其他分享
首页 > 其他分享> > pytorch+CUDA10.0+pycharm

pytorch+CUDA10.0+pycharm

作者:互联网

先安装Anaconda3

清华大学的源,我装的最新版本2020.11

https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

安装pycharm

装免费的就行了community版本

https://www.jetbrains.com/pycharm/download/#section=windows

安装CUDA

打开Nvidia控制面板->帮助->系统信息->组件->3D设置 看一下显卡最高支持CUDA多少,一般按照最高的来就行,我的是CUDA10.0
https://developer.nvidia.com/cuda-toolkit-archive
这里能找到对应的CUDA安装包,对应的下载安装即可

安装pytorch

打开anaconda
首先换成清华源

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/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

然后创建虚拟环境,我这个版本conda支持python3.8

conda create -n pytorch python=3.8

完成后,去pytorch网站,因为我是CUDA10.0,可能找不到相应版本的CUDA,没关系,改一下就可以了
https://pytorch.org/

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

把后面的 -c pytorch去掉,因为这个是指在官网下载,我们是要在清华源下载,这样自己就能安装了,收工~

最后在pycharm里面,创建的工程里面,配置python的来源就选择到…/anaconda/env/python.exe即可

标签:--,edu,pytorch,conda,https,anaconda,pycharm,CUDA10.0
来源: https://blog.csdn.net/guet_gjl/article/details/110240867