torch安装

第n次安装了,每次都得结合很多篇才能安装成功,这次直接自己写一个。

一、安装anaconda

这个自己搜吧,挺简单的。

二、安装pytorch

进入anaconda prompt

watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBAbTBfNTk0NzY4OTk=,size_16,color_FFFFFF,t_70,g_se,x_16

创建一个虚拟环境(就不用再安装python了,创建之后直接就是用的3.8版本的python)

conda create -n env_name python=3.8

进入虚拟环境

conda activate env_name

使用清华镜像源(使用http而不是https,参考这篇博客

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

安装pytorch(官网

  • 不要用pip并去掉-c python的方法(会默认下载cpu),用conda直接复制run this command(其实挺快的);
  • 如果电脑没有gpu那就不要选cuda11.3,选cpu 
  • solve环境之后看看装的是不是cuda版本,实在不行用查一查版本号用pip安装
  • watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBAbTBfNTk0NzY4OTk=,size_20,color_FFFFFF,t_70,g_se,x_16
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

三、验证

89f9575b048a43748b3162a4d4735e24.png

没报错,而且是True(电脑没有gpu的小伙伴就是false啦) 

四、镜像源安装其他库

  • scikit-learn

pip install sklearn -i https://pypi.tuna.tsinghua.edu.cn/simple 
  • pandas
pip install pandas -i https://mirrors.aliyun.com/pypi/simple/ --trusted-host=mirrors.aliyun.com/pypi/simple
  • rdkit
conda install -c conda-forge rdkit
  • 有setup.py的安装文件
cd 文件夹_name
python setup.py install
  • torch-geometric

先安装依赖库,torch_cluster  ; torch_scatter  ; torch_sparse  ;   torch_spline_conv

conda list查看pytorch版本与cuda版本,我的是1.10.2+cu113

4f70ee8cbd9642f389d36b5a0cba9412.png

 在官网下载依赖库,注意python版本号不要下载错误

082c040f23d74e8193938daabbd942fb.png

再安装torch_geometric

#可能会有点慢,报一堆warning
conda install torch_geometric

安装GPU版本的Torch可以通过以下步骤进行: 1. 首先,确保你的计算机上已经安装了NVIDIA显卡驱动程序,并且支持CUDA。你可以在NVIDIA官方网站上下载并安装最新的显卡驱动程序。 2. 接下来,安装CUDA工具包。你可以在NVIDIA官方网站上下载适合你显卡的CUDA版本,并按照官方文档的指导进行安装。 3. 安装cuDNN库。cuDNN是一个针对深度神经网络的GPU加速库,可以提高Torch在GPU上的性能。你可以在NVIDIA官方网站上下载适合你CUDA版本的cuDNN库,并按照官方文档的指导进行安装。 4. 确保你的Python环境已经安装了pip包管理器。然后使用以下命令安装GPU版本的Torchtorchvision: ``` pip install torch torchvision ``` 5. 安装完成后,你可以使用以下命令验证Torch是否成功安装并可用: ``` python import torch print(torch.cuda.is_available()) ``` 如果输出为True,则表示Torch已经成功安装并可以在GPU上运行了。 请注意,以上步骤是基于使用pip包管理器进行安装的。如果你使用的是conda包管理器,可以使用类似的命令进行安装,但可能会有一些细微的差异。 #### 引用[.reference_title] - *1* *2* *3* [Cuda 和 GPU版torch安装最全攻略,以及在GPU 上运行 torch代码](https://blog.csdn.net/weixin_46833029/article/details/127946022)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值